tf.raw_ops.UnbatchGrad
Градиент операции Unbatch.
tf.raw_ops.UnbatchGrad(
original_input,
batch_index,
grad,
id,
container='',
shared_name='',
name=None
)
Работает как Batch, но использует указанный индекс batch_index для пакетной обработки элементов по мере их появления. Это гарантирует, что градиенты будут переданы обратно в том же сеансе, в котором выполнялся прямой проход.
original_input: входные данные операции Unbatch, градиентом которых является это значение. batch_index: значение batch_index, переданное операции Unbatch, градиентом которой является это значение. grad: градиент последующего слоя. id: скалярный идентификатор, выданный Batch. batched_grad: возвращаемое значение — пустой тензор или пакетный градиент. container: контейнер для управления совместным использованием ресурсов. shared_name: предполагается, что экземпляры UnbatchGrad с одинаковыми container и shared_name могут относиться к одному пакету. Если значение не задано, в качестве shared name будет использоваться имя операции.
| Аргументы | |
|---|---|
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 Tensor. Has the same type as original_input. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/raw_ops/UnbatchGrad