tensorflow::ops::SparseApplyRMSProp
#include <training_ops.h>
Обновление '*var' согласно алгоритму RMSProp.
Краткое описание
Обратите внимание, что в плотной реализации этого алгоритма ms и mom будут обновляться даже если grad равен нулю, но в этой разреженной реализации ms и mom не будут обновляться в итерациях, где grad равен нулю.
mean_square = decay * mean_square + (1-decay) * gradient ** 2 Delta = learning_rate * gradient / sqrt(mean_square + epsilon)
Аргументы:
- scope: Объект Scope
- var: Должен быть получен из Variable().
- ms: Должен быть получен из Variable().
- mom: Должен быть получен из Variable().
- lr: Коэффициент масштабирования. Должен быть скаляром.
- rho: Скорость затухания. Должен быть скаляром.
- epsilon: Член Ridge. Должен быть скаляром.
- grad: Градиент.
- indices: Вектор индексов в первом измерении var, ms и mom.
Дополнительные атрибуты (см. Attrs):
- use_locking: Если
True, обновление тензоров var, ms и mom защищено блокировкой; в противном случае поведение не определено, но может характеризоваться меньшим числом конфликтов.
Возвращаемое значение:
-
Output: То же самое, что и "var".
| Конструкторы и деструкторы | |
|---|---|
SparseApplyRMSProp(const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input ms, ::tensorflow::Input mom, ::tensorflow::Input lr, ::tensorflow::Input rho, ::tensorflow::Input momentum, ::tensorflow::Input epsilon, ::tensorflow::Input grad, ::tensorflow::Input indices) | |
SparseApplyRMSProp(const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input ms, ::tensorflow::Input mom, ::tensorflow::Input lr, ::tensorflow::Input rho, ::tensorflow::Input momentum, ::tensorflow::Input epsilon, ::tensorflow::Input grad, ::tensorflow::Input indices, const SparseApplyRMSProp::Attrs & attrs) |
| Публичные атрибуты | |
|---|---|
operation | |
out | |
| Публичные функции | |
|---|---|
node() const | ::tensorflow::Node * |
operator::tensorflow::Input() const | |
operator::tensorflow::Output() const | |
| Публичные статические функции | |
|---|---|
UseLocking(bool x) | |
| Структуры | |
|---|---|
| tensorflow::ops::SparseApplyRMSProp::Attrs | Дополнительные атрибуты для SparseApplyRMSProp. |
Публичные атрибуты
operation
Operation operation
out
::tensorflow::Output out
Публичные функции
SparseApplyRMSProp
SparseApplyRMSProp( const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input ms, ::tensorflow::Input mom, ::tensorflow::Input lr, ::tensorflow::Input rho, ::tensorflow::Input momentum, ::tensorflow::Input epsilon, ::tensorflow::Input grad, ::tensorflow::Input indices )
SparseApplyRMSProp
SparseApplyRMSProp( const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input ms, ::tensorflow::Input mom, ::tensorflow::Input lr, ::tensorflow::Input rho, ::tensorflow::Input momentum, ::tensorflow::Input epsilon, ::tensorflow::Input grad, ::tensorflow::Input indices, const SparseApplyRMSProp::Attrs & attrs )
node
::tensorflow::Node * node() const
operator::tensorflow::Input
operator::tensorflow::Input() const
operator::tensorflow::Output
operator::tensorflow::Output() const
Публичные статические функции
UseLocking
Attrs UseLocking( bool x )
За исключением случаев, оговоренных в ином порядке, содержимое этой страницы лицензировано по лицензии Creative Commons Attribution 4.0, а примеры кода лицензированы по лицензии Apache 2.0. Для получения дополнительной информации см. Политики веб-сайта Google для разработчиков. Java является зарегистрированным товарным знаком Oracle и/или её аффилированных компаний.
Последнее обновление 2020-06-29 UTC.
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/cc/class/tensorflow/ops/sparse-apply-r-m-s-prop