org.tensorflow.op.core

Class RaggedGather<T>

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.core.RaggedGather<T>

Type Parameters:

T - data type for outputDenseValues() output

All Implemented Interfaces:

Op

public final class RaggedGather<T>
extends PrimitiveOp

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

Field Summary

Fields inherited from class org.tensorflow.op.PrimitiveOp

operation

Method Summary

Modifier and Type	Method and Description
static <T,U extends Number> RaggedGather<T>	create(Scope scope, Iterable<Operand<Long>> paramsNestedSplits, Operand<T> paramsDenseValues, Operand<U> indices, Long OUTPUTRAGGEDRANK) Factory method to create a class to wrap a new RaggedGather operation to the graph.
Output<T>	outputDenseValues() The `flat_values` for the returned RaggedTensor.
List<Output<Long>>	outputNestedSplits() The `nested_row_splits` tensors that define the row-partitioning for the returned RaggedTensor.

Methods inherited from class org.tensorflow.op.PrimitiveOp

equals, hashCode, toString

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Method Detail

create

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

Factory method to create a class to wrap a new RaggedGather operation to the graph.

Parameters:

scope - current graph 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

outputNestedSplits

public List<Output<Long>> outputNestedSplits()

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

outputDenseValues

public Output<T> outputDenseValues()

The `flat_values` for the returned RaggedTensor.