тензорный поток:: опс:: TensorArrayGradWithShape
#include <data_flow_ops.h>
Создает TensorArray для хранения нескольких градиентов значений в данном дескрипторе.
Краткое содержание
Аналогично TensorArrayGradV3. Однако он создает аккумулятор расширенной формы по сравнению с входным TensorArray , градиент которого вычисляется. Это позволяет рассчитывать несколько градиентов для одного и того же TensorArray с использованием одного и того же аккумулятора.
Аргументы:
- область: объект области.
- handle: Дескриптор прямого TensorArray .
- flow_in: скаляр с плавающей точкой, обеспечивающий правильную цепочку операций.
- shape_to_prepend: вектор int32, представляющий фигуру. Элементы в аккумуляторе градиента будут иметь форму, которая представляет собой значение shape_to_prepend, объединенное с формой элементов в TensorArray , соответствующих дескриптору ввода.
- source: исходная строка градиента, используемая для принятия решения о том, какой градиент TensorArray возвращать.
Возврат:
Конструкторы и деструкторы | |
---|---|
TensorArrayGradWithShape (const :: tensorflow::Scope & scope, :: tensorflow::Input handle, :: tensorflow::Input flow_in, :: tensorflow::Input shape_to_prepend, StringPiece source) |
Публичные атрибуты | |
---|---|
flow_out | |
grad_handle | |
operation |
Публичные атрибуты
поток_выход
::tensorflow::Output flow_out
grad_handle
::tensorflow::Output grad_handle
операция
Operation operation
Общественные функции
TensorArrayGradWithShape
TensorArrayGradWithShape( const ::tensorflow::Scope & scope, ::tensorflow::Input handle, ::tensorflow::Input flow_in, ::tensorflow::Input shape_to_prepend, StringPiece source )