graph2mat.bindings.torch.modules.matrixblock
Torch wrappers for matrix block.
Classes
|
Wrapper for matrix block to make it use torch instead of numpy. |
- class graph2mat.bindings.torch.modules.matrixblock.TorchMatrixBlock(i_basis: PointBasis, j_basis: PointBasis, operation_cls: Type, symm_transpose: bool = False, preprocessor=None, **operation_kwargs)[source]
Bases:
MatrixBlock
,Module
Wrapper for matrix block to make it use torch instead of numpy.
- numpy: ModuleType = <module 'torch' from '/opt/hostedtoolcache/Python/3.11.11/x64/lib/python3.11/site-packages/torch/__init__.py'>