dgl.sparse.SparseMatrix.csr

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

返回稀疏矩阵的压缩稀疏行（CSR）表示。

See CSR in Wikipedia.

此函数还返回一个索引张量作为值索引，表示CSR表示中非零元素值的顺序。一个None值索引数组表示值的顺序与SparseMatrix的值保持不变。

Returns:

  • torch.Tensor – 行索引指针

  • torch.Tensor – 列索引

  • torch.Tensor – 值索引

示例

>>> 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]))