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

CSRSparseMatrixComponents

public final class CSRSparseMatrixComponents

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.

Public Methods

Output<Integer>
colInds()
An array containing CSR matrix column indices.
static <T> CSRSparseMatrixComponents<T>
create(Scope scope, Operand<?> csrSparseMatrix, Operand<Integer> index, Class<T> type)
Factory method to create a class wrapping a new CSRSparseMatrixComponents operation.
Output<Integer>
rowPtrs()
An array containing CSR matrix row pointers.
Output<T>
values()
An array containing CSR matrix nonzero values.

Inherited Methods

Public Methods

public Output<Integer> colInds ()

An array containing CSR matrix column indices.

public static CSRSparseMatrixComponents<T> create (Scope scope, Operand<?> csrSparseMatrix, Operand<Integer> index, Class<T> type)

Factory method to create a class wrapping a new CSRSparseMatrixComponents operation.

Parameters
scope current scope
csrSparseMatrix A batched CSRSparseMatrix.
index The index in `csr_sparse_matrix`'s batch.
Returns
  • a new instance of CSRSparseMatrixComponents

public Output<Integer> rowPtrs ()

An array containing CSR matrix row pointers.

public Output<T> values ()

An array containing CSR matrix nonzero values.