SparseSplit

Stay organized with collections Save and categorize content based on your preferences.
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>>
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

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.