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. Дескриптор forward TensorArray. |
flow_in | A Tensor типа float32. Скаляр с плавающей точкой, который обеспечивает правильное цепное выполнение операций. |
shape_to_prepend | A Tensor типа int32. Вектор int32, представляющий форму. Элементы в накопителе градиента будут иметь форму, которая является результатом конкатенации значения shape_to_prepend с формой элементов в TensorArray, соответствующих входному дескриптору. |
source | Строка источника градиента, используемая для определения того, какой градиент 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/versions/r2.9/api_docs/python/tf/raw_ops/TensorArrayGradWithShape
