Split a `SparseTensor` into `num_split` tensors along one dimension.
If the `shape[split_dim]` is not an integer multiple of `num_split`. Slices `[0 : shape[split_dim] % num_split]` gets one extra dimension. For example, if `split_dim = 1` and `num_split = 2` and the input is
input_tensor = shape = [2, 7] [ a d e ] [b c ]
Graphically the output tensors are:
output_tensor[0] = shape = [2, 4] [ a ] [b c ]
output_tensor[1] = shape = [2, 3] [ d e ] [ ]
Constants
String | OP_NAME | The name of this op, as known by TensorFlow core engine |
Public Methods
static <T extends TType> SparseSplit<T> | |
List<Output<TInt64>> | |
List<Output<TInt64>> |
outputShape()
A list of 1-D tensors represents the shape of the output sparse
tensors.
|
List<Output<T>> |
outputValues()
A list of 1-D tensors represents the values of the output sparse
tensors.
|
Inherited Methods
Constants
public static final String OP_NAME
The name of this op, as known by TensorFlow core engine
Public Methods
public static SparseSplit<T> create (Scope scope, Operand<TInt64> splitDim, Operand<TInt64> indices, Operand<T> values, Operand<TInt64> shape, Long numSplit)
Factory method to create a class wrapping a new SparseSplit operation.
Parameters
scope | current scope |
---|---|
splitDim | 0-D. The dimension along which to split. Must be in the range `[0, rank(shape))`. |
indices | 2-D tensor represents the indices of the sparse tensor. |
values | 1-D tensor represents the values of the sparse tensor. |
shape | 1-D. tensor represents the shape of the sparse tensor. output indices: A list of 1-D tensors represents the indices of the output sparse tensors. |
numSplit | The number of ways to split. |
Returns
- a new instance of SparseSplit
public List<Output<TInt64>> outputShape ()
A list of 1-D tensors represents the shape of the output sparse tensors.
public List<Output<T>> outputValues ()
A list of 1-D tensors represents the values of the output sparse tensors.