dgl.sparse.div

dgl.sparse.div(A: SparseMatrix, B: SparseMatrix | Number | Tensor) → SparseMatrix[源码]

对 SparseMatrix 进行逐元素除法，等同于 A / B。

如果 A 和 B 都是稀疏矩阵，它们都必须是对角矩阵。

参数:

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

返回值:

稀疏矩阵

返回值类型:

SparseMatrix

示例

>>> A = dglsp.diag(torch.arange(1, 4))
>>> B = dglsp.diag(torch.arange(10, 13))
>>> dglsp.div(A, B)
SparseMatrix(indices=tensor([[0, 1, 2],
                             [0, 1, 2]]),
             values=tensor([0.1000, 0.1818, 0.2500]),
             shape=(3, 3), nnz=3)

>>> A = dglsp.diag(torch.arange(1, 4))
>>> dglsp.div(A, 2)
SparseMatrix(indices=tensor([[0, 1, 2],
                             [0, 1, 2]]),
             values=tensor([0.5000, 1.0000, 1.5000]),
             shape=(3, 3), nnz=3)

>>> indices = torch.tensor([[1, 0, 2], [0, 3, 2]])
>>> val = torch.tensor([1, 2, 3])
>>> A = dglsp.spmatrix(indices, val, shape=(3, 4))
>>> dglsp.div(A, 2)
SparseMatrix(indices=tensor([[1, 0, 2],
                             [0, 3, 2]]),
             values=tensor([0.5000, 1.0000, 1.5000]),
             shape=(3, 4), nnz=3)