dgl.sparse.spspmm

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

将一个稀疏矩阵乘以另一个稀疏矩阵，等同于 A @ B。

两个稀疏矩阵的非零值必须是 1D 的。

参数：

• A (SparseMatrix) – 形状为 (L, M) 的稀疏矩阵

• B (SparseMatrix) – 形状为 (M, N) 的稀疏矩阵

返回值：

形状为 (L, N) 的稀疏矩阵。

返回类型：

SparseMatrix

示例

>>> indices1 = torch.tensor([[0, 1, 1], [1, 0, 1]])
>>> val1 = torch.ones(len(row1))
>>> A = dglsp.spmatrix(indices1, val1)
>>> indices2 = torch.tensor([[0, 1, 1], [0, 2, 1]])
>>> val2 = torch.ones(len(row2))
>>> B = dglsp.spmatrix(indices2, val2)
>>> dglsp.spspmm(A, B)
SparseMatrix(indices=tensor([[0, 0, 1, 1, 1],
                             [1, 2, 0, 1, 2]]),
             values=tensor([1., 1., 1., 1., 1.]),
             shape=(2, 3), nnz=5)