dgl.sparse.SparseMatrix.csc

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

返回稀疏矩阵的压缩稀疏列 (CSC) 表示。

参见 Wikipedia 上的 CSC。

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

返回值：

• torch.Tensor – 列指针 (indptr)

• torch.Tensor – 行索引

• torch.Tensor – 值索引

示例

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