dgl.sparse.add

dgl.sparse.add(A: SparseMatrix, B: SparseMatrix) → SparseMatrix

对 SparseMatrix 进行逐元素相加，等价于 A + B。

参数:

• A (SparseMatrix) – 稀疏矩阵

• B (SparseMatrix) – 稀疏矩阵

返回:

稀疏矩阵

返回类型:

SparseMatrix

示例

>>> indices = torch.tensor([[1, 0, 2], [0, 1, 2]])
>>> val = torch.tensor([10, 20, 30])
>>> A = dglsp.spmatrix(indices, val)
>>> B = dglsp.diag(torch.arange(1, 4))
>>> dglsp.add(A, B)
SparseMatrix(indices=tensor([[0, 0, 1, 1, 2],
                             [0, 1, 0, 1, 2]]),
             values=tensor([1, 20, 10,  2, 33]),
             shape=(3, 3), nnz=5)