operation
Modifier and Type | Method and Description |
---|---|
static <T,U extends Number> |
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.
|
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]`.
|
equals, hashCode, op, toString
clone, finalize, getClass, notify, notifyAll, wait, wait, wait
forEach, spliterator
public static <T,U extends Number> SplitV<T> create(Scope scope, Operand<T> value, Operand<U> sizeSplits, Operand<Integer> axis, Long numSplit)
scope
- current scopevalue
- 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))`.numSplit
- public List<Output<T>> output()
Copyright © 2022. All rights reserved.