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.