тензорный поток:: опс:: ПрименитьДобавитьСигн

#include <training_ops.h>

Обновите «*var» в соответствии с обновлением AddSign.

Краткое содержание

m_t <- beta1 * m_{t-1} + (1 - beta1) * g update <- (alpha +sign_decay *sign(g) *sign(m)) * g переменная <- переменная - lr_t * update

Аргументы:

  • область: объект области.
  • var: Должно быть из переменной().
  • m: Должно быть из переменной().
  • lr: коэффициент масштабирования. Должно быть скаляр.
  • альфа: Должно быть скаляр.
  • Sign_decay: Должен быть скаляром.
  • бета: Должен быть скаляром.
  • град: Градиент.

Необязательные атрибуты (см. Attrs ):

  • use_locking: если True , обновление тензоров var и m защищено блокировкой; в противном случае поведение не определено, но может вызывать меньше конфликтов.

Возврат:

  • Output : То же, что и «var».

Конструкторы и деструкторы

ApplyAddSign (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input m, :: tensorflow::Input lr, :: tensorflow::Input alpha, :: tensorflow::Input sign_decay, :: tensorflow::Input beta, :: tensorflow::Input grad)
ApplyAddSign (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input m, :: tensorflow::Input lr, :: tensorflow::Input alpha, :: tensorflow::Input sign_decay, :: tensorflow::Input beta, :: tensorflow::Input grad, const ApplyAddSign::Attrs & attrs)

Публичные атрибуты

operation
out

Общественные функции

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Публичные статические функции

UseLocking (bool x)

Структуры

tensorflow::ops::ApplyAddSign::Attrs

Дополнительные установщики атрибутов для ApplyAddSign .

Публичные атрибуты

операция 

Operation operation

вне 

::tensorflow::Output out

Общественные функции

ПрименитьДобавитьСигн 

 ApplyAddSign(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input var,
  ::tensorflow::Input m,
  ::tensorflow::Input lr,
  ::tensorflow::Input alpha,
  ::tensorflow::Input sign_decay,
  ::tensorflow::Input beta,
  ::tensorflow::Input grad
)

ПрименитьДобавитьСигн 

 ApplyAddSign(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input var,
  ::tensorflow::Input m,
  ::tensorflow::Input lr,
  ::tensorflow::Input alpha,
  ::tensorflow::Input sign_decay,
  ::tensorflow::Input beta,
  ::tensorflow::Input grad,
  const ApplyAddSign::Attrs & attrs
)

узел 

::tensorflow::Node * node() const

оператор::tensorflow::Input 

 operator::tensorflow::Input() const

оператор::tensorflow::Выход 

 operator::tensorflow::Output() const

Публичные статические функции

Использование блокировки 

Attrs UseLocking(
  bool x
)