dgl.sparse.val_like

dgl.sparse.val_like(mat: SparseMatrix, val: Tensor) → SparseMatrix

使用新值从现有稀疏矩阵创建稀疏矩阵。

新的稀疏矩阵将具有与给定稀疏矩阵相同的非零索引，并使用给定值作为新的非零值。

参数:

• mat (SparseMatrix) – 具有非零值的现有稀疏矩阵

• val (torch.Tensor) – 非零元素的新值，形状为 (nnz) 或 (nnz, D) 的张量

返回:

新的稀疏矩阵

返回类型:

SparseMatrix

示例

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