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

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()
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()
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.