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. Вещественная скалярная величина, которая обеспечивает правильную цепочку операций. |
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. |
© 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.3/api_docs/python/tf/raw_ops/TensorArrayGradWithShape