Spec-Zone.ru › TensorFlow

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 с одинаковым контейнером и shared_name могут принадлежать одной группе. Если оставлено пустым, имя операции будет использовано в качестве имени shared.

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

Spec-Zone.ru

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