tensorflow::ops::ResourceApplyAdamWithAmsgrad
#include <training_ops.h>
Обновляет '*var' согласно алгоритму Adam.
Сводка
$$\text{lr}_t := \mathrm{learning_rate} * \sqrt{1 - \beta_2^t} / (1 - \beta_1^t)$$ $$m_t := \beta_1 * m_{t-1} + (1 - \beta_1) * g$$ $$v_t := \beta_2 * v_{t-1} + (1 - \beta_2) * g * g$$ $$\hat{v}_t := max{\hat{v}_{t-1}, v_t}$$ $$\text{variable} := \text{variable} - \text{lr}_t * m_t / (\sqrt{\hat{v}_t} + \epsilon)$$
Аргументы:
- scope: объект Scope
- var: Должна быть получена из Variable().
- m: Должна быть получена из Variable().
- v: Должна быть получена из Variable().
- vhat: Должна быть получена из Variable().
- beta1_power: Должно быть скалярным значением.
- beta2_power: Должно быть скалярным значением.
- lr: Коэффициент масштабирования. Должен быть скалярным значением.
- beta1: Коэффициент импульса. Должен быть скалярным значением.
- beta2: Коэффициент импульса. Должен быть скалярным значением.
- epsilon: Параметр регуляризации. Должен быть скалярным значением.
- grad: Градиент.
Необязательные атрибуты (см. Attrs):
- use_locking: Если
True, обновление тензоров var, m и v будет защищено блокировкой; в противном случае поведение не определено, но может снизить конкуренцию за ресурсы.
Возвращает:
- созданный
Operation
| Конструкторы и деструкторы | |
|---|---|
ResourceApplyAdamWithAmsgrad(const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input m, ::tensorflow::Input v, ::tensorflow::Input vhat, ::tensorflow::Input beta1_power, ::tensorflow::Input beta2_power, ::tensorflow::Input lr, ::tensorflow::Input beta1, ::tensorflow::Input beta2, ::tensorflow::Input epsilon, ::tensorflow::Input grad) | |
ResourceApplyAdamWithAmsgrad(const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input m, ::tensorflow::Input v, ::tensorflow::Input vhat, ::tensorflow::Input beta1_power, ::tensorflow::Input beta2_power, ::tensorflow::Input lr, ::tensorflow::Input beta1, ::tensorflow::Input beta2, ::tensorflow::Input epsilon, ::tensorflow::Input grad, const ResourceApplyAdamWithAmsgrad::Attrs & attrs) |
| Открытые атрибуты | |
|---|---|
operation | |
| Открытые функции | |
|---|---|
operator::tensorflow::Operation() const | |
| Открытые статические функции | |
|---|---|
UseLocking(bool x) | |
| Структуры | |
|---|---|
| tensorflow::ops::ResourceApplyAdamWithAmsgrad::Attrs | Методы установки необязательных атрибутов для ResourceApplyAdamWithAmsgrad. |
Открытые атрибуты
operation
Operation operation
Открытые функции
ResourceApplyAdamWithAmsgrad
ResourceApplyAdamWithAmsgrad( const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input m, ::tensorflow::Input v, ::tensorflow::Input vhat, ::tensorflow::Input beta1_power, ::tensorflow::Input beta2_power, ::tensorflow::Input lr, ::tensorflow::Input beta1, ::tensorflow::Input beta2, ::tensorflow::Input epsilon, ::tensorflow::Input grad )
ResourceApplyAdamWithAmsgrad
ResourceApplyAdamWithAmsgrad( const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input m, ::tensorflow::Input v, ::tensorflow::Input vhat, ::tensorflow::Input beta1_power, ::tensorflow::Input beta2_power, ::tensorflow::Input lr, ::tensorflow::Input beta1, ::tensorflow::Input beta2, ::tensorflow::Input epsilon, ::tensorflow::Input grad, const ResourceApplyAdamWithAmsgrad::Attrs & attrs )
operator::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-adam-with-amsgrad