torch.sparse_compressed_tensor
torch.sparse_compressed_tensor(compressed_indices, plain_indices, values, size=None, *, dtype=None, layout=None, device=None, requires_grad=False, check_invariants=None) -> Tensor
Constructs a sparse tensor in Compressed Sparse format - CSR, CSC,
   BSR, or BSC - with specified values at the given
   "compressed_indices" and "plain_indices". Sparse matrix
   multiplication operations in Compressed Sparse format are typically
   faster than that for sparse tensors in COO format. Make you have a
   look at the note on the data type of the indices.
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:
      * compressed_indices (array_like) -- (B+1)-dimensional