dgl.sparse.spspmm

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

将稀疏矩阵与稀疏矩阵相乘，等同于 A @ B。

两个稀疏矩阵的非零值必须是一维的。

Parameters:

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

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

Returns:

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

Return type:

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)