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

Inherited Methods

From class org.tensorflow.op.PrimitiveOp

final boolean	equals(Object obj)
final int	hashCode()
Operation	op() Returns the underlying Operation
final String	toString()

From class java.lang.Object

boolean	equals(Object arg0)
final Class<?>	getClass()
int	hashCode()
final void	notify()
final void	notifyAll()
String	toString()
final void	wait(long arg0, int arg1)
final void	wait(long arg0)
final void	wait()

From interface java.lang.Iterable

void	forEach(Consumer<? super T> arg0)
abstract Iterator<Operand<T>>	iterator()
Spliterator<Operand<T>>	spliterator()

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