graph2mat.bindings.torch
Interface with pytorch.
The interface with torch is simple to understand. Instead of
using numpy we need to use torch.
We wrap the BasisMatrixData and Graph2Mat to do that, generating
TorchBasisMatrixData and TorchGraph2Mat respectively. Any framework
that uses torch should use these classes instead of the original ones.
Also, if extra bindings are needed for a framework that uses torch,
they should take these bindings as a starting point. E.g. if bindings
are implemented for X, XGraph2Mat should inherit from TorchGraph2Mat.
These bindings contain no extra functionality, as all that we do is to
make sure that the core functionality works with torch tensors.
Modules
Wrappers for data handling in pytorch. |
|
Wrappers for graph2mat modules in pytorch. |