rllm.utils.sparse_mx_to_torch_sparse_tensor¶

class rllm.utils.sparse_mx_to_torch_sparse_tensor(sparse_mx)[source]¶

Bases:

Convert a scipy sparse matrix to a torch.sparse.Tensor.

Parameters:: sparse_mx (scipy.sparse.spmatrix) – The input scipy sparse matrix.
Returns:: A sparse COO tensor with float32 values.
Return type:: Tensor