dgl.sparse.SparseMatrix.coalesce

SparseMatrix.coalesce()

返回一个合并的稀疏矩阵。

合并后的稀疏矩阵满足以下属性：

• 非零元素的索引是唯一的，

• 索引按字典序排序。

合并过程将通过求和累积相同索引的非零元素。

此函数不支持自动梯度计算 (autograd)。

返回：

合并后的稀疏矩阵

返回类型：

SparseMatrix

示例

>>> indices = torch.tensor([[1, 0, 0, 0, 1], [1, 1, 1, 2, 2]])
>>> val = torch.tensor([0, 1, 2, 3, 4])
>>> A = dglsp.spmatrix(indices, val)
>>> A.coalesce()
SparseMatrix(indices=tensor([[0, 0, 1, 1],
                             [1, 2, 1, 2]]),
             values=tensor([3, 3, 0, 4]),
             shape=(2, 3), nnz=4)