dgl.sparse.spmm

dgl.sparse.spmm(A: SparseMatrix, X: Tensor) → Tensor

将稀疏矩阵与稠密矩阵相乘，等同于 A @ X。

参数:

• A (SparseMatrix) – 形状为 (L, M) 的稀疏矩阵，带标量值

• X (torch.Tensor) – 形状为 (M, N) 或 (M) 的稠密矩阵

返回值:

形状为 (L, N) 或 (L) 的稠密矩阵

返回类型:

torch.Tensor

示例

>>> indices = torch.tensor([[0, 1, 1], [1, 0, 1]])
>>> val = torch.randn(indices.shape[1])
>>> A = dglsp.spmatrix(indices, val)
>>> X = torch.randn(2, 3)
>>> result = dglsp.spmm(A, X)
>>> type(result)
<class 'torch.Tensor'>
>>> result.shape
torch.Size([2, 3])