cugraph_pyg.tensor.DistMatrix#
- class cugraph_pyg.tensor.DistMatrix(
- src: Tuple[Tensor, Tensor] | Tuple[DistTensor, DistTensor] | str | List[str] | None = None,
- shape: list | tuple | None = None,
- dtype: dtype | None = None,
- device: Literal['cpu', 'cuda'] | None = 'cpu',
- backend: Literal['nccl', 'vmm'] | None = 'nccl',
- format: Literal['csr', 'csc', 'coo'] | None = 'coo',
WholeGraph-backed Distributed Matrix Interface for PyTorch.
- Attributes:
- dtype
- local_col
- local_coo
- local_row
- shape
Methods
transpose([view])Return the transpose of this matrix.
get_local_tensor
- __init__(
- src: Tuple[Tensor, Tensor] | Tuple[DistTensor, DistTensor] | str | List[str] | None = None,
- shape: list | tuple | None = None,
- dtype: dtype | None = None,
- device: Literal['cpu', 'cuda'] | None = 'cpu',
- backend: Literal['nccl', 'vmm'] | None = 'nccl',
- format: Literal['csr', 'csc', 'coo'] | None = 'coo',
Methods
__init__([src, shape, dtype, device, ...])get_local_tensor()transpose([view])Return the transpose of this matrix.
Attributes
dtypelocal_collocal_coolocal_rowshape