dgl.sparse.SparseMatrix.neg

SparseMatrix.neg() SparseMatrix

返回一个新稀疏矩阵,其非零值为原始非零值的相反数,等同于 -A

返回

稀疏矩阵的相反数

返回类型

SparseMatrix

示例

>>> indices = torch.tensor([[1, 1, 3], [1, 2, 3]])
>>> val = torch.tensor([1., 1., 2.])
>>> A = dglsp.spmatrix(indices, val)
>>> A = -A
SparseMatrix(indices=tensor([[1, 1, 3],
                             [1, 2, 3]]),
             values=tensor([-1., -1., -2.]),
             shape=(4, 4), nnz=3)