org.tensorflow.op.core

Class Split<T>

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.core.Split<T>

Type Parameters:

T - data type for output() output

All Implemented Interfaces:

Iterable<Operand<T>>, Op

@Operator
public final class Split<T>
extends PrimitiveOp
implements Iterable<Operand<T>>

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

Field Summary

Fields inherited from class org.tensorflow.op.PrimitiveOp

operation

Method Summary

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 to wrap a new Split operation to the graph.
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`.

Methods inherited from class org.tensorflow.op.PrimitiveOp

equals, hashCode, toString

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Methods inherited from interface java.lang.Iterable

forEach, spliterator

Method Detail

create

public static <T> Split<T> create(Scope scope,
                                  Operand<Integer> axis,
                                  Operand<T> value,
                                  Long numSplit)

Factory method to create a class to wrap a new Split operation to the graph.

Parameters:

scope - current graph 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

output

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

iterator

public Iterator<Operand<T>> iterator()

Specified by:

iterator in interface Iterable<Operand<T>>