operation
Modifier and Type | Method and Description |
---|---|
static <T> Split<T> |
create(Scope scope,
Operand<Integer> axis,
Operand<T> value,
Long numSplit)
Factory method to create a class wrapping a new Split operation.
|
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`.
|
equals, hashCode, op, toString
clone, finalize, getClass, notify, notifyAll, wait, wait, wait
forEach, spliterator
public static <T> Split<T> create(Scope scope, Operand<Integer> axis, Operand<T> value, Long numSplit)
scope
- current scopeaxis
- 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]`.public List<Output<T>> output()
Copyright © 2022. All rights reserved.