Spec-Zone.ru › TensorFlow

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, который использует (нецентрированный) второй момент. Это часто помогает в обучении, но требует немного больше вычислительных ресурсов и памяти.

Обратите внимание, что в плотной (dense) реализации этого алгоритма mg, ms и mom будут обновляться, даже если grad равен нулю, но в этой разреженной (sparse) реализации 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 Tensor типа resource. Должен быть получен из Variable().
mg Tensor типа resource. Должен быть получен из Variable().
ms Tensor типа resource. Должен быть получен из Variable().
mom Tensor типа resource. Должен быть получен из Variable().
lr Tensor. Должен быть одного из следующих типов: float32, float64, int32, uint8, int16, int8, complex64, int64, qint8, quint8, qint32, bfloat16, qint16, quint16, uint16, complex128, half, uint32, uint64. Масштабирующий коэффициент. Должен быть скаляром.
rho Tensor. Должен иметь тот же тип, что и lr. Скорость затухания. Должен быть скаляром.
momentum Tensor. Должен иметь тот же тип, что и lr. Масштаб импульса (momentum). Должен быть скаляром.
epsilon Tensor. Должен иметь тот же тип, что и lr. Гребневый член (ridge term). Должен быть скаляром.
grad Tensor. Должен иметь тот же тип, что и lr. Градиент.
use_locking Необязательный bool. По умолчанию False. Если True, обновление тензоров var, mg, ms и mom защищено блокировкой; в противном случае поведение не определено, но может наблюдаться меньшая состязательность (contention).
name Имя для операции (необязательно).
Возвращает
Созданный Operation.

© 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/python/tf/raw_ops/ResourceApplyCenteredRMSProp

Spec-Zone.ru

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