tf.raw_ops.UnbatchGrad

Gradient of Unbatch.

View aliases

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.raw_ops.UnbatchGrad

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.