dgl.sparse.SparseMatrix.double

SparseMatrix.double()

将矩阵值转换为 double 数据类型。如果矩阵已使用 double 数据类型，则返回原始矩阵。

返回:

带有 double 值的矩阵

返回类型:

SparseMatrix

示例

>>> indices = torch.tensor([[1, 1, 2], [1, 2, 0]])
>>> A = dglsp.spmatrix(indices, shape=(3, 4))
>>> A.double()
SparseMatrix(indices=tensor([[1, 1, 2],
                             [1, 2, 0]]),
             values=tensor([1., 1., 1.], dtype=torch.float64),
             shape=(3, 4), nnz=3)