SparseBsrTensor

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

Abstract Signature:

SparseBsrTensor(crow_indices: Tensor, col_indices: Tensor, values: Tensor, size: Optional[List[int]], dtype: Optional[str], device: Optional[Any])

PyTorchJAX (Core)TensorFlow

PyTorch

API: torch.sparse_bsr_tensor

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Plugin (sparse_bsr_constructor)

TensorFlow

API: —

Strategy: Custom / Partial