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. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/raw_ops/UnbatchGrad