tensorflow :: ops :: ApplyCenteredRMSProp

#include <training_ops.h>

Perbarui '* var' sesuai dengan algoritme RMSProp yang terpusat.

Ringkasan

Algoritma RMSProp terpusat menggunakan perkiraan momen kedua yang terpusat (yaitu, varians) untuk normalisasi, sebagai lawan dari RMSProp biasa, yang menggunakan momen kedua (tidak terpusat). Ini sering membantu pelatihan, tetapi sedikit lebih mahal dalam hal komputasi dan memori.

Perhatikan bahwa dalam implementasi padat dari algoritme ini, mg, ms, dan mom akan memperbarui meskipun gradnya nol, tetapi dalam implementasi yang jarang ini, mg, ms, dan mom tidak akan memperbarui dalam iterasi yang nilainya nol.

mean_square = decay * mean_square + (1-decay) * gradient ** 2 mean_grad = decay * mean_grad + (1-decay) * gradien

Delta = learning_rate * gradien / sqrt (mean_square + epsilon - mean_grad ** 2)

mg <- rho * mg_ {t-1} + (1-rho) * grad ms <- rho * ms_ {t-1} + (1-rho) * grad * grad mom <- momentum * mom_ {t-1 } + lr * grad / sqrt (ms - mg * mg + epsilon) var <- var - mom

Argumen:

  • scope: Objek Scope
  • var: Harus dari Variabel ().
  • mg: Harus dari Variabel ().
  • ms: Harus dari Variabel ().
  • ibu: Harus dari Variabel ().
  • lr: Faktor penskalaan. Harus skalar.
  • rho: Tingkat kerusakan. Harus skalar.
  • epsilon: Istilah punggungan. Harus skalar.
  • grad: Gradien.

Atribut opsional (lihat Attrs ):

  • use_locking: Jika True , pembaruan tensor var, mg, ms, dan mom dilindungi oleh kunci; jika tidak, perilakunya tidak ditentukan, tetapi mungkin menunjukkan lebih sedikit perselisihan.

Pengembalian:

  • Output : Sama seperti "var".

Pembuat dan Penghancur

ApplyCenteredRMSProp (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input mg, :: tensorflow::Input ms, :: tensorflow::Input mom, :: tensorflow::Input lr, :: tensorflow::Input rho, :: tensorflow::Input momentum, :: tensorflow::Input epsilon, :: tensorflow::Input grad)
ApplyCenteredRMSProp (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input mg, :: tensorflow::Input ms, :: tensorflow::Input mom, :: tensorflow::Input lr, :: tensorflow::Input rho, :: tensorflow::Input momentum, :: tensorflow::Input epsilon, :: tensorflow::Input grad, const ApplyCenteredRMSProp::Attrs & attrs)

Atribut publik

operation
out

Fungsi publik

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

Fungsi statis publik

UseLocking (bool x)

Structs

tensorflow :: ops :: ApplyCenteredRMSProp :: Attrs

Penyetel atribut opsional untuk ApplyCenteredRMSProp .

Atribut publik

operasi

Operation operation

di luar

::tensorflow::Output out

Fungsi publik

ApplyCenteredRMSProp

 ApplyCenteredRMSProp(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input var,
  ::tensorflow::Input mg,
  ::tensorflow::Input ms,
  ::tensorflow::Input mom,
  ::tensorflow::Input lr,
  ::tensorflow::Input rho,
  ::tensorflow::Input momentum,
  ::tensorflow::Input epsilon,
  ::tensorflow::Input grad
)

ApplyCenteredRMSProp

 ApplyCenteredRMSProp(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input var,
  ::tensorflow::Input mg,
  ::tensorflow::Input ms,
  ::tensorflow::Input mom,
  ::tensorflow::Input lr,
  ::tensorflow::Input rho,
  ::tensorflow::Input momentum,
  ::tensorflow::Input epsilon,
  ::tensorflow::Input grad,
  const ApplyCenteredRMSProp::Attrs & attrs
)

simpul

::tensorflow::Node * node() const 

operator :: tensorflow :: Input

 operator::tensorflow::Input() const 

operator :: tensorflow :: Keluaran

 operator::tensorflow::Output() const 

Fungsi statis publik

UseLocking

Attrs UseLocking(
  bool x
)