tensorflow::ops::TensorArrayGradWithShape
#include <data_flow_ops.h>Создает TensorArray для хранения нескольких градиентов значений в заданном дескрипторе.
Краткое содержание
Аналогично TensorArrayGradV3. Однако он создает накопитель с расширенной формой по сравнению с входным TensorArray, градиент которого вычисляется. Это позволяет вычислять несколько градиентов для одного и того же TensorArray, используя один и тот же накопитель.
Аргументы:
- scope: Объект Scope
- handle: Дескриптор прямого массива TensorArray .
- flow_in: Скалярное значение типа float, обеспечивающее корректную цепочку операций.
- 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 | |
Общедоступные атрибуты
flow_out
::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 )