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]`.