graph2mat.bindings.torch.modules.matrixblock

Torch wrappers for matrix block.

Classes

TorchMatrixBlock(i_basis, j_basis, operation_cls)

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'>