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.

Constants

String OP_NAME The name of this op, as known by TensorFlow core engine

Public Methods

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

Inherited Methods

Constants

public static final String OP_NAME

The name of this op, as known by TensorFlow core engine

Constant Value: "CSRSparseMatrixComponents"

Public Methods

public Output<TInt32> colInds ()

An array containing CSR matrix column indices.

public static CSRSparseMatrixComponents<T> create (Scope scope, Operand<?> csrSparseMatrix, Operand<TInt32> 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<TInt32> rowPtrs ()

An array containing CSR matrix row pointers.

public Output<T> values ()

An array containing CSR matrix nonzero values.