tf.raw_ops.UnbatchGrad | TensorFlow v2.16.1 (original) (raw)

tf.raw_ops.UnbatchGrad

Stay organized with collections Save and categorize content based on your preferences.

Gradient of Unbatch.

View aliases

Compat aliases for migration

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.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.

Last updated 2024-04-26 UTC.