Have a question? Connect with the community at the TensorFlow Forum Visit Forum

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