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.
| Аргументы | |
|---|---|
original_input | Tensor. |
batch_index | Tensor типа int64. |
grad | Tensor. Должен иметь тот же тип, что и original_input. |
id | Tensor типа int64. |
container | Необязательный string. По умолчанию "". |
shared_name | Необязательный string. По умолчанию "". |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
Tensor. Имеет тот же тип, что и 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.4/api_docs/python/tf/raw_ops/UnbatchGrad