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> | |
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 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