public final class
Split
Splits a tensor into `num_split` tensors along one dimension.
Public Methods
static <T> Split<T> | |
Iterator<Operand<T>> |
iterator()
|
List<Output<T>> |
output()
They are identically shaped tensors, whose shape matches that of `value`
except along `axis`, where their sizes are
`values.shape[split_dim] / num_split`.
|
Inherited Methods
Public Methods
public static Split<T> create (Scope scope, Operand<Integer> axis, Operand<T> value, Long numSplit)
Factory method to create a class wrapping a new Split operation.
Parameters
scope | current scope |
---|---|
axis | 0-D. The dimension along which to split. Must be in the range `[-rank(value), rank(value))`. |
value | The tensor to split. |
numSplit | The number of ways to split. Must evenly divide `value.shape[split_dim]`. |
Returns
- a new instance of Split
public List<Output<T>> output ()
They are identically shaped tensors, whose shape matches that of `value` except along `axis`, where their sizes are `values.shape[split_dim] / num_split`.