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

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