dgl.sparse.mul

dgl.sparse.mul(A: SparseMatrix | Number | Tensor, B: SparseMatrix | Number | Tensor) → SparseMatrix[source]

对 SparseMatrix 进行逐元素乘法，等同于 A * B。

如果 A 和 B 都是稀疏矩阵，则它们都应为对角矩阵。

参数:

A (SparseMatrix 或标量) – 稀疏矩阵或标量值
B (SparseMatrix 或标量) – 稀疏矩阵或标量值

返回:

稀疏矩阵

返回类型:

SparseMatrix

示例

>>> indices = torch.tensor([[1, 0, 2], [0, 3, 2]])
>>> val = torch.tensor([10, 20, 30])
>>> A = dglsp.spmatrix(indices, val)
>>> dglsp.mul(A, 2)
SparseMatrix(indices=tensor([[1, 0, 2],
                             [0, 3, 2]]),
             values=tensor([20, 40, 60]),
             shape=(3, 4), nnz=3)

>>> D = dglsp.diag(torch.arange(1, 4))
>>> dglsp.mul(D, 2)
SparseMatrix(indices=tensor([[0, 1, 2],
                             [0, 1, 2]]),
             values=tensor([2, 4, 6]),
             shape=(3, 3), nnz=3)

>>> D = dglsp.diag(torch.arange(1, 4))
>>> dglsp.mul(D, D)
SparseMatrix(indices=tensor([[0, 1, 2],
                             [0, 1, 2]]),
             values=tensor([1, 4, 9]),
             shape=(3, 3), nnz=3)