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>	flowOut()
Output<?>	gradHandle()

Inherited Methods

From class org.tensorflow.op.PrimitiveOp

final boolean	equals(Object obj)
final int	hashCode()
Operation	op() Returns the underlying 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()

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