Spec-Zone.ru › TensorFlow 2.4

tf.raw_ops.UnbatchGrad

Градиент Unbatch.

Просмотр псевдонимов

Псевдонимы для миграции

См. Руководство по миграции для получения дополнительных сведений.

tf.compat.v1.raw_ops.UnbatchGrad

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

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API