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 | Строка источника градиента, используемая для определения, какой градиент 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.4/api_docs/python/tf/raw_ops/TensorArrayGradWithShape