SparseCompressedTensor

Constructs a sparse tensor in a compressed format (CSR, CSC, BSR, BSC).

Abstract Signature:

SparseCompressedTensor(compressed_indices: Tensor, plain_indices: Tensor, values: Tensor, size: Optional[List[int]], layout: Optional[Any])

PyTorchJAX (Core)

PyTorch

API: torch.sparse_compressed_tensor

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Plugin (sparse_compressed_constructor)