pygmtools.pytorch_backend

Functions

build_batch	Pytorch implementation of building a batched tensor
dense_to_sparse	Pytorch implementation of converting a dense adjacency matrix to a sparse matrix
from_numpy	Pytorch function from_numpy
gaussian_aff_fn	Pytorch implementation of Gaussian affinity function
hungarian	Pytorch implementation of Hungarian algorithm
inner_prod_aff_fn	Pytorch implementation of inner product affinity function
ipfp	Pytorch implementation of IPFP algorithm
rrwm	Pytorch implementation of RRWM algorithm.
sinkhorn	Pytorch implementation of Sinkhorn algorithm
sm	Pytorch implementation of SM algorithm.
to_numpy	Pytorch function to_numpy