ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more


Reads out the CSR components at batch index.

This op is meant only for debugging / testing, and its interface is not expected to be stable.

csr_sparse_matrix A Tensor of type variant. A batched CSRSparseMatrix.
index A Tensor of type int32. The index in csr_sparse_matrix's batch.
type A tf.DType from: tf.float32, tf.float64, tf.complex64, tf.complex128.
name A name for the operation (optional).

A tuple of Tensor objects (row_ptrs, col_inds, values).
row_ptrs A Tensor of type int32.
col_inds A Tensor of type int32.
values A Tensor of type type.