torch.sparse_coo_tensor
torch.sparse_coo_tensor(indices, values, size=None, *, dtype=None, device=None, requires_grad=False, check_invariants=None) -> Tensor
Constructs a sparse tensor in COO(rdinate) format with specified
   values at the given "indices".
Note:
 This function returns an uncoalesced tensor.

Note:
 If the "device" argument is not specified the device of the given
 "values" and indices tensor(s) must match. If, however, the
 argument is specified the input Tensors will be converted to the
 given device and in turn determine the device of the constructed
 sparse tensor.

Parameters:
      * indices (array_like) -- Initial data for the tensor. Can
        be a list, tuple, NumPy "ndarray", scalar, and other types.
        Will be cast to a "torch.LongTensor" internally. The indices
        are the coordinates of the non-zero values in the matrix, and
        thus should be two-dimensional where the first dimension is