dgl.sparse.power

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

对 SparseMatrix 进行逐元素幂运算，等价于 A ** scalar。

参数:

A (SparseMatrix) – 稀疏矩阵
scalar (Scalar) – 指数

返回值:

稀疏矩阵

返回类型:

SparseMatrix

示例

>>> indices = torch.tensor([[1, 0, 2], [0, 3, 2]])
>>> val = torch.tensor([10, 20, 30])
>>> A = dglsp.spmatrix(indices, val)
>>> dglsp.power(A, 2)
SparseMatrix(indices=tensor([[1, 0, 2],
                             [0, 3, 2]]),
             values=tensor([100, 400, 900]),
             shape=(3, 4), nnz=3)

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