public final class
SplitV
Splits a tensor into `num_split` tensors along one dimension.
Public Methods
static <T, U extends Number> SplitV<T> | |
Iterator<Operand<T>> |
iterator()
|
List<Output<T>> |
output()
Tensors whose shape matches that of `value`
except along `axis`, where their sizes are
`size_splits[i]`.
|
Inherited Methods
Public Methods
public static SplitV<T> create (Scope scope, Operand<T> value, Operand<U> sizeSplits, Operand<Integer> axis, Long numSplit)
Factory method to create a class wrapping a new SplitV operation.
Parameters
scope | current scope |
---|---|
value | The tensor to split. |
sizeSplits | list containing the sizes of each output tensor along the split dimension. Must sum to the dimension of value along split_dim. Can contain one -1 indicating that dimension is to be inferred. |
axis | 0-D. The dimension along which to split. Must be in the range `[-rank(value), rank(value))`. |
Returns
- a new instance of SplitV
public List<Output<T>> output ()
Tensors whose shape matches that of `value` except along `axis`, where their sizes are `size_splits[i]`.