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

Аргументы
`original_input`	`Tensor`.
`batch_index`	`Tensor` типа `int64`.
`grad`	`Tensor`. Должен иметь тот же тип, что и `original_input`.
`id`	`Tensor` типа `int64`.
`container`	Необязательный `string`. По умолчанию `""`.
`shared_name`	Необязательный `string`. По умолчанию `""`.
`name`	Имя операции (необязательно).

Возвращаемое значение
`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/versions/r2.9/api_docs/python/tf/raw_ops/UnbatchGrad