ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

TensorArrayGather

public final class TensorArrayGather

Gather specific elements from the TensorArray into output `value`.

All elements selected by `indices` must have the same shape.

Nested Classes

class TensorArrayGather.Options Optional attributes for TensorArrayGather  

Public Methods

Output<T>
asOutput()
Returns the symbolic handle of a tensor.
static <T> TensorArrayGather<T>
create(Scope scope, Operand<?> handle, Operand<Integer> indices, Operand<Float> flowIn, Class<T> dtype, Options... options)
Factory method to create a class wrapping a new TensorArrayGather operation.
static TensorArrayGather.Options
elementShape(Shape elementShape)
Output<T>
value()
All of the elements in the TensorArray, concatenated along a new axis (the new dimension 0).

Inherited Methods

Public Methods

public Output<T> asOutput ()

Returns the symbolic handle of a tensor.

Inputs to TensorFlow operations are outputs of another TensorFlow operation. This method is used to obtain a symbolic handle that represents the computation of the input.

public static TensorArrayGather<T> create (Scope scope, Operand<?> handle, Operand<Integer> indices, Operand<Float> flowIn, Class<T> dtype, Options... options)

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

Parameters
scope current scope
handle The handle to a TensorArray.
indices The locations in the TensorArray from which to read tensor elements.
flowIn A float scalar that enforces proper chaining of operations.
dtype The type of the elem that is returned.
options carries optional attributes values
Returns
  • a new instance of TensorArrayGather

public static TensorArrayGather.Options elementShape (Shape elementShape)

Parameters
elementShape The expected shape of an element, if known. Used to validate the shapes of TensorArray elements. If this shape is not fully specified, gathering zero-size TensorArrays is an error.

public Output<T> value ()

All of the elements in the TensorArray, concatenated along a new axis (the new dimension 0).