Spec-Zone.ru › TensorFlow 2.9

tf.raw_ops.ResourceApplyCenteredRMSProp

Обновление '*var' согласно алгоритму RMSProp с центрированием.

Просмотр псевдонимов

Псевдонимы совместимости для миграции

Для получения дополнительной информации см. руководство по миграции.

tf.compat.v1.raw_ops.ResourceApplyCenteredRMSProp

tf.raw_ops.ResourceApplyCenteredRMSProp(
    var,
    mg,
    ms,
    mom,
    lr,
    rho,
    momentum,
    epsilon,
    grad,
    use_locking=False,
    name=None
)

Алгоритм 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 * 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

Аргументы
var Переменная типа resource. Должна быть из Variable().
mg Переменная типа resource. Должна быть из Variable().
ms Переменная типа resource. Должна быть из Variable().
mom Переменная типа resource. Должна быть из Variable().
lr Значение. Должно быть одного из следующих типов: float32, float64, int32, uint8, int16, int8, complex64, int64, qint8, quint8, qint32, bfloat16, uint16, complex128, half, uint32, uint64. Коэффициент масштабирования. Должен быть скаляром.
rho Значение. Должно иметь тот же тип, что и lr. Скорость затухания. Должен быть скаляром.
momentum Значение. Должно иметь тот же тип, что и lr. Масштаб импульса. Должен быть скаляром.
epsilon Значение. Должно иметь тот же тип, что и lr. Слагаемое ridge. Должен быть скаляром.
grad Значение. Должно иметь тот же тип, что и lr. Градиент.
use_locking Необязательный параметр. По умолчанию False. Если True, обновление тензоров var, mg, ms и mom защищено блокировкой; в противном случае поведение не определено, но может иметь меньше конфликтов.
name Имя операции (необязательно).
Возвращаемое значение
Созданная операция.

© 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/versions/r2.9/api_docs/python/tf/raw_ops/ResourceApplyCenteredRMSProp

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API