Gradient of Unbatch.
Acts like Batch but using the given batch_index index of batching things as they become available. This ensures that the gradients are propagated back in the same session which did the forward pass.
original_input: The input to the Unbatch operation this is the gradient of. batch_index: The batch_index given to the Unbatch operation this is the gradient of. grad: The downstream gradient. id: The id scalar emitted by Batch. batched_grad: The return value, either an empty tensor or the batched gradient. container: Container to control resource sharing. shared_name: Instances of UnbatchGrad with the same container and shared_name are assumed to possibly belong to the same batch. If left empty, the op name will be used as the shared name.
Nested Classes
| class | UnbatchGrad.Options | Optional attributes for UnbatchGrad
|
|
Constants
| String | OP_NAME | The name of this op, as known by TensorFlow core engine |
Public Methods
| Output<T> |
asOutput()
Returns the symbolic handle of the tensor.
|
| Output<T> | |
| static UnbatchGrad.Options |
container(String container)
|
| static <T extends TType> UnbatchGrad<T> | |
| static UnbatchGrad.Options |
sharedName(String sharedName)
|
Inherited Methods
Constants
public static final String OP_NAME
The name of this op, as known by TensorFlow core engine
Public Methods
public Output<T> asOutput ()
Returns the symbolic handle of the 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 UnbatchGrad<T> create (Scope scope, Operand<T> originalInput, Operand<TInt64> batchIndex, Operand<T> grad, Operand<TInt64> id, Options... options)
Factory method to create a class wrapping a new UnbatchGrad operation.
Parameters
| scope | current scope |
|---|---|
| options | carries optional attributes values |
Returns
- a new instance of UnbatchGrad