torch_npu.npu_apply_adam

API接口

torch_npu.npu_apply_adam(beta1_power, beta2_power, lr, beta1, beta2, epsilon, grad, use_locking, use_nesterov, out = (var, m, v))

功能描述

adam结果计数。

参数说明

• beta1_power (Scalar) - beta1的幂。
• beta2_power (Scalar) - beta2的幂。
• lr (Scalar) - 学习率。
• beta1 (Scalar) - 一阶矩估计值的指数衰减率。
• beta2 (Scalar) - 二阶矩估计值的指数衰减率。
• epsilon (Scalar) - 添加到分母中以提高数值稳定性的项数。
• grad (Tensor) - 梯度。
• use_locking (Bool，可选) - 设置为True时使用lock进行更新操作。
• use_nesterov (Bool，可选) - 设置为True时采用nesterov更新。
• var (Tensor) - 待优化变量。
• m (Tensor) - 变量平均值。
• v (Tensor) - 变量方差。