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

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
Returns
  • a new instance of TensorArrayGradWithShape

public Output<Float> flowOut ()

public Output<?> gradHandle ()