SparseCsrTensor

Constructs a sparse tensor in CSR (Compressed Sparse Row) format.

Abstract Signature:

SparseCsrTensor(crow_indices: Tensor, col_indices: Tensor, values: Tensor, size: Optional[List[int]])

PyTorchJAX (Core)

PyTorch

API: torch.sparse_csr_tensor

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: jax.experimental.sparse.BCSR

Strategy: Plugin (sparse_csr_constructor)