SparseSplit

public final class SparseSplit

Split a `SparseTensor` into `num_split` tensors along one dimension.

If the `shape[split_dim]` is not an integer multiple of `num_split`. Slices `[0 : shape[split_dim] % num_split]` gets one extra dimension. For example, if `split_dim = 1` and `num_split = 2` and the input is

input_tensor = shape = [2, 7] [ a d e ] [b c ]

Graphically the output tensors are:

output_tensor[0] = shape = [2, 4] [ a ] [b c ]

output_tensor[1] = shape = [2, 3] [ d e ] [ ]

Constants

String OP_NAME The name of this op, as known by TensorFlow core engine

Public Methods

static <T extends TType> SparseSplit<T>	create(Scope scope, Operand<TInt64> splitDim, Operand<TInt64> indices, Operand<T> values, Operand<TInt64> shape, Long numSplit) Factory method to create a class wrapping a new SparseSplit operation.
List<Output<TInt64>>	outputIndices()
List<Output<TInt64>>	outputShape() A list of 1-D tensors represents the shape of the output sparse tensors.
List<Output<T>>	outputValues() A list of 1-D tensors represents the values of the output sparse tensors.

Inherited Methods

From class org.tensorflow.op.RawOp

final boolean	equals(Object obj)
final int	hashCode()
Operation	op() Return this unit of computation as a single `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 org.tensorflow.op.Op

abstract ExecutionEnvironment	env() Return the execution environment this op was created in.
abstract Operation	op() Return this unit of computation as a single `Operation`.

Constants

public static final String OP_NAME

The name of this op, as known by TensorFlow core engine

Constant Value: "SparseSplit"

Public Methods

public static SparseSplit<T> create (Scope scope, Operand<TInt64> splitDim, Operand<TInt64> indices, Operand<T> values, Operand<TInt64> shape, Long numSplit)

Factory method to create a class wrapping a new SparseSplit operation.

Parameters

scope	current scope
splitDim	0-D. The dimension along which to split. Must be in the range `[0, rank(shape))`.
indices	2-D tensor represents the indices of the sparse tensor.
values	1-D tensor represents the values of the sparse tensor.
shape	1-D. tensor represents the shape of the sparse tensor. output indices: A list of 1-D tensors represents the indices of the output sparse tensors.
numSplit	The number of ways to split.

Returns

a new instance of SparseSplit

public List<Output<TInt64>> outputIndices ()

public List<Output<TInt64>> outputShape ()

A list of 1-D tensors represents the shape of the output sparse tensors.

public List<Output<T>> outputValues ()

A list of 1-D tensors represents the values of the output sparse tensors.