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.