RaggedGather

public final class RaggedGather

Gather ragged slices from `params` axis `0` according to `indices`.

Outputs a `RaggedTensor` output composed from `output_dense_values` and `output_nested_splits`, such that: 

output.shape = indices.shape + params.shape[1:]
 output.ragged_rank = indices.shape.ndims + params.ragged_rank
 output[i...j, d0...dn] = params[indices[i...j], d0...dn]
where
  • `params = ragged.from_nested_row_splits(params_dense_values, params_nested_splits)` provides the values that should be gathered.
  • `indices` ia a dense tensor with dtype `int32` or `int64`, indicating which values should be gathered.
  • `output = ragged.from_nested_row_splits(output_dense_values, output_nested_splits)` is the output tensor.
(Note: This c++ op is used to implement the higher-level python `tf.ragged.gather` op, which also supports ragged indices.)

Constants

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

Public Methods

static <T extends TNumber, U extends TType> RaggedGather<T, U>
create(Scope scope, Iterable<Operand<T>> paramsNestedSplits, Operand<U> paramsDenseValues, Operand<? extends TNumber> indices, Long OUTPUTRAGGEDRANK)
Factory method to create a class wrapping a new RaggedGather operation.
Output<U>
outputDenseValues()
The `flat_values` for the returned RaggedTensor.
List<Output<T>>
outputNestedSplits()
The `nested_row_splits` tensors that define the row-partitioning for the returned RaggedTensor.

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: "RaggedGather"

Public Methods

public static RaggedGather<T, U> create (Scope scope, Iterable<Operand<T>> paramsNestedSplits, Operand<U> paramsDenseValues, Operand<? extends TNumber> indices, Long OUTPUTRAGGEDRANK)

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

Parameters
scope current scope
paramsNestedSplits The `nested_row_splits` tensors that define the row-partitioning for the `params` RaggedTensor input.
paramsDenseValues The `flat_values` for the `params` RaggedTensor. There was a terminology change at the python level from dense_values to flat_values, so dense_values is the deprecated name.
indices Indices in the outermost dimension of `params` of the values that should be gathered.
OUTPUTRAGGEDRANK The ragged rank of the output RaggedTensor. `output_nested_splits` will contain this number of `row_splits` tensors. This value should equal `indices.shape.ndims + params.ragged_rank - 1`.
Returns
  • a new instance of RaggedGather

public Output<U> outputDenseValues ()

The `flat_values` for the returned RaggedTensor.

public List<Output<T>> outputNestedSplits ()

The `nested_row_splits` tensors that define the row-partitioning for the returned RaggedTensor.