Split

public final class Split

Splits a tensor into `num_split` tensors along one dimension.

Public Methods

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>>
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 Iterator<Operand<T>> iterator ()

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