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