dgl.sparse.SparseMatrix.float

SparseMatrix.float()[source]

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

Returns

The matrix with float values

Return type

SparseMatrix

Examples

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