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

TensorArrayGradWithShape

public final class TensorArrayGradWithShape

Creates a TensorArray for storing multiple gradients of values in the given handle.

Similar to TensorArrayGradV3. However it creates an accumulator with an expanded shape compared to the input TensorArray whose gradient is being computed. This enables multiple gradients for the same TensorArray to be calculated using the same accumulator.

Public Methods

static TensorArrayGradWithShape
create(Scope scope, Operand<?> handle, Operand<Float> flowIn, Operand<Integer> shapeToPrepend, String source)
Factory method to create a class wrapping a new TensorArrayGradWithShape operation.
Output<Float>
Output<?>

Inherited Methods

Public Methods

public static TensorArrayGradWithShape create (Scope scope, Operand<?> handle, Operand<Float> flowIn, Operand<Integer> shapeToPrepend, String source)

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

Parameters
scope current scope
handle The handle to the forward TensorArray.
flowIn A float scalar that enforces proper chaining of operations.
shapeToPrepend An int32 vector representing a shape. Elements in the gradient accumulator will have shape which is this shape_to_prepend value concatenated with shape of the elements in the TensorArray corresponding to the input handle.
source The gradient source string, used to decide which gradient TensorArray to return.
Returns
  • a new instance of TensorArrayGradWithShape

public Output<Float> flowOut ()

public Output<?> gradHandle ()