Spec-Zone.ru › TensorFlow

tf.raw_ops.ApplyAdaMax

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

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

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

См. Руководство по миграции для получения дополнительных сведений.

tf.compat.v1.raw_ops.ApplyAdaMax

tf.raw_ops.ApplyAdaMax(
    var,
    m,
    v,
    beta1_power,
    lr,
    beta1,
    beta2,
    epsilon,
    grad,
    use_locking=False,
    name=None
)

mt <- beta1 * m{t-1} + (1 - beta1) * g vt <- max(beta2 * v{t-1}, abs(g)) variable <- variable - learning_rate / (1 - beta1^t) * m_t / (v_t + epsilon)

Аргументы
var Изменяемая Tensor. Должна быть одного из следующих типов: float32, float64, int32, uint8, int16, int8, complex64, int64, qint8, quint8, qint32, bfloat16, qint16, quint16, uint16, complex128, half, uint32, uint64. Должна быть получена из Variable().
m Изменяемая Tensor. Должна иметь тот же тип, что и var. Должна быть получена из Variable().
v Изменяемая Tensor. Должна иметь тот же тип, что и var. Должна быть получена из Variable().
beta1_power Tensor. Должна иметь тот же тип, что и var. Должна быть скаляром.
lr Tensor. Должна иметь тот же тип, что и var. Коэффициент масштабирования. Должна быть скаляром.
beta1 Tensor. Должна иметь тот же тип, что и var. Коэффициент импульса. Должна быть скаляром.
beta2 Tensor. Должна иметь тот же тип, что и var. Коэффициент импульса. Должна быть скаляром.
epsilon Tensor. Должна иметь тот же тип, что и var. Слагаемое Риджа. Должна быть скаляром.
grad Tensor. Должна иметь тот же тип, что и var. Градиент.
use_locking Необязательный bool. По умолчанию False. Если True, обновление тензоров var, m и v будет защищено блокировкой; в противном случае поведение не определено, но может проявлять меньше конкуренции.
name Имя операции (необязательно).
Возвращаемое значение
Изменяемая Tensor. Имеет тот же тип, что и var.

© 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/ApplyAdaMax

Spec-Zone.ru

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