pygmtools.pytorch_backendΒΆ
Functions
Pytorch implementation of building a batched tensor |
|
Pytorch implementation of CAO solver in fast config (mode="pc") |
|
Pytorch implementation of CAO solver (mode="c") |
|
Pytorch implementation of computing affinity score |
|
Pytorch implementation of converting a dense adjacency matrix to a sparse matrix |
|
Pytorch function from_numpy |
|
Pytorch implementation of Graduated Assignment for Multi-Graph Matching (with compatibility for 2GM and clustering) |
|
The real forward function of GAMGM |
|
Pytorch implementation of Gaussian affinity function |
|
Pytorch implementation of generate_isomorphic_graphs |
|
Pytorch implementation of Hungarian algorithm |
|
Pytorch implementation of inner product affinity function |
|
Pytorch implementation of IPFP algorithm |
|
Pytorch implementation of permutation_loss |
|
Pytorch implementation of RRWM algorithm. |
|
Pytorch implementation of Sinkhorn algorithm |
|
Pytorch implementation of SM algorithm. |
|
Pytorch function to_numpy |
Classes
Torch wrapper to support forward and backward pass (by black-box differentiation) |