tf.raw_ops.TensorArrayGradWithShape
Создает TensorArray для хранения нескольких градиентов значений в заданном дескрипторе.
tf.raw_ops.TensorArrayGradWithShape(
handle, flow_in, shape_to_prepend, source, name=None
)
Похоже на TensorArrayGradV3. Однако эта операция создает аккумулятор с расширенной формой по сравнению с входным TensorArray, градиент которого вычисляется. Это позволяет вычислять несколько градиентов для одного и того же TensorArray с помощью одного аккумулятора.
| Аргументы | |
|---|---|
handle | A Tensor типа resource. Дескриптор прямого TensorArray. |
flow_in | A Tensor типа float32. Скаляр типа float, обеспечивающий правильное связывание операций в цепочку. |
shape_to_prepend | A Tensor типа int32. Вектор int32, представляющий форму. Элементы аккумулятора градиентов будут иметь форму, полученную конкатенацией значения shape_to_prepend и формы элементов TensorArray, соответствующего входному дескриптору. |
source | A string. Строка источника градиента, используемая для определения того, какой TensorArray градиентов вернуть. |
name | Имя операции (необязательно). |
| Возвращает | |
|---|---|
Кортеж объектов Tensor (grad_handle, flow_out). | |
grad_handle | A Tensor типа resource. |
flow_out | A Tensor типа float32. |
© 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/TensorArrayGradWithShape