dgl.sparse.SparseMatrix.long

SparseMatrix.long()[source]

Converts the matrix values to long data type. If the matrix already uses long data type, the original matrix will be returned.

Returns:

The matrix with long values

Return type:

DiagMatrix

Examples

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