tensorflow::ops::ResourceApplyCenteredRMSProp
#include <training_ops.h>
Обновление '*var' согласно алгоритму RMSProp с центрированием.
Краткое описание
Алгоритм RMSProp с центрированием использует оценку центрированного второго момента (т.е. дисперсии) для нормализации, в отличие от обычного RMSProp, который использует (нецентрированный) второй момент. Это часто помогает в процессе обучения, но немного дороже в вычислительном отношении и по потреблению памяти.
Обратите внимание, что в плотной реализации этого алгоритма mg, ms и mom будут обновляться, даже если grad равен нулю, но в разреженной реализации mg, ms и mom не будут обновляться в итерациях, в которых grad равен нулю.
mean_square = decay * mean_square + (1-decay) * gradient ** 2 mean_grad = decay * mean_grad + (1-decay) * gradient
Delta = learning_rate * gradient / sqrt(mean_square + epsilon - mean_grad ** 2)
mg <- rho * mgt-1 + (1-rho) * grad ms <- rho * mst-1 + (1-rho) * grad * grad mom <- momentum * momt-1 + lr * grad / sqrt(ms - mg * mg + epsilon) var <- var - mom
Аргументы:
- scope: Объект Scope
- var: Должен быть получен из Variable().
- mg: Должен быть получен из Variable().
- ms: Должен быть получен из Variable().
- mom: Должен быть получен из Variable().
- lr: Коэффициент масштабирования. Должен быть скалярным.
- rho: Скорость затухания. Должен быть скалярным.
- momentum: Масштаб импульса. Должен быть скалярным.
- epsilon: Слагаемое для сглаживания. Должен быть скалярным.
- grad: Градиент.
Дополнительные атрибуты (см. Attrs):
- use_locking: Если
True, обновление тензоров var, mg, ms и mom защищено блокировкой; в противном случае поведение не определено, но может проявлять меньшую конкуренцию.
Возвращаемое значение:
- созданный
Operation
| Конструкторы и деструкторы | |
|---|---|
ResourceApplyCenteredRMSProp(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) | |
ResourceApplyCenteredRMSProp(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 ResourceApplyCenteredRMSProp::Attrs & attrs) |
| Общедоступные атрибуты | |
|---|---|
operation | |
| Общедоступные функции | |
|---|---|
operator::tensorflow::Operation() const | |
| Общедоступные статические функции | |
|---|---|
UseLocking(bool x) | |
| Структуры | |
|---|---|
| tensorflow::ops::ResourceApplyCenteredRMSProp::Attrs | Дополнительные атрибуты-установщики для ResourceApplyCenteredRMSProp. |
Общедоступные атрибуты
operation
Operation operation
Общедоступные функции
ResourceApplyCenteredRMSProp
ResourceApplyCenteredRMSProp( 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 )
ResourceApplyCenteredRMSProp
ResourceApplyCenteredRMSProp( 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 ResourceApplyCenteredRMSProp::Attrs & attrs )
Оператор::tensorflow::Operation
operator::tensorflow::Operation() const
Общедоступные статические функции
UseLocking
Attrs UseLocking( bool x )
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/cc/class/tensorflow/ops/resource-apply-centered-r-m-s-prop