Gradient of Unbatch.
tf.raw_ops.UnbatchGrad(
original_input, batch_index, grad, id, container='', shared_name='', name=None
)
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.
Args | |
|---|---|
original_input
|
A Tensor.
|
batch_index
|
A Tensor of type int64.
|
grad
|
A Tensor. Must have the same type as original_input.
|
id
|
A Tensor of type int64.
|
container
|
An optional string. Defaults to "".
|
shared_name
|
An optional string. Defaults to "".
|
name
|
A name for the operation (optional). |
Returns | |
|---|---|
A Tensor. Has the same type as original_input.
|