dgl.sparse.SparseMatrix.csr

SparseMatrix.csr() → Tuple[Tensor, Tensor, Tensor]

返回稀疏矩阵的压缩稀疏行 (CSR) 表示。

参见 维基百科中的 CSR。

此函数还返回值索引作为索引张量，指示 CSR 表示中非零元素值的顺序。值为 None 的值索引数组表示值的顺序与 SparseMatrix 的值相同。

返回:

• torch.Tensor – 行 indptr

• torch.Tensor – 列 indices

• torch.Tensor – 值 indices

示例

>>> indices = torch.tensor([[1, 2, 1], [2, 4, 3]])
>>> A = dglsp.spmatrix(indices)
>>> A.csr()
(tensor([0, 0, 2, 3]), tensor([2, 3, 4]), tensor([0, 2, 1]))