dmlc--dgl
Hongzhi (Steve), Chen
aabbcee29c
* add_backslash * fix * more * blabla Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
31 行
785 B
Python
31 行
785 B
Python
"""DGL unary operators for sparse matrix module."""
|
|
from .sparse_matrix import SparseMatrix, val_like
|
|
|
|
|
|
def neg(A: SparseMatrix) -> SparseMatrix:
|
|
"""Returns a new sparse matrix with the negation of the original nonzero
|
|
values, equivalent to ``-A``.
|
|
|
|
Returns
|
|
-------
|
|
SparseMatrix
|
|
Negation of the sparse matrix
|
|
|
|
Examples
|
|
--------
|
|
|
|
>>> 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)
|
|
"""
|
|
return val_like(A, -A.val)
|
|
|
|
|
|
SparseMatrix.neg = neg
|
|
SparseMatrix.__neg__ = neg
|