Function at::_fused_adam_out(at::TensorList, at::TensorList, at::TensorList, at::TensorList, at::TensorList, at::TensorList, at::TensorList, double, double, double, double, double, bool, bool, const ::std::optional<at::Tensor>&, const ::std::optional<at::Tensor>&)¶

Function Documentation¶

inline void at::_fused_adam_out(at::TensorList out, at::TensorList self, at::TensorList grads, at::TensorList exp_avgs, at::TensorList exp_avg_sqs, at::TensorList max_exp_avg_sqs, at::TensorList state_steps, double lr, double beta1, double beta2, double weight_decay, double eps, bool amsgrad, bool maximize, const ::std::optional<at::Tensor> &grad_scale = {}, const ::std::optional<at::Tensor> &found_inf = {})¶