torch.optim.adamax.adamax

torch.optim.adamax.adamax(params, grads, exp_avgs, exp_infs, state_steps, foreach=None, maximize=False, differentiable=False, capturable=False, has_complex=False, *, eps, beta1, beta2, lr, weight_decay)

执行 Adamax 算法计算的功能性 API。

有关详细信息，请参阅 Adamax。