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

• Defined in File Functions.h

Function Documentation

inline ::std::tuple<::std::vector<at::Tensor>, ::std::vector<at::Tensor>, ::std::vector<at::Tensor>> at::_fused_sgd(at::TensorList self, at::TensorList grads, at::TensorList momentum_buffer_list, double weight_decay, double momentum, double lr, double dampening, bool nesterov, bool maximize, bool is_first_step, const ::std::optional<at::Tensor> &grad_scale = {}, const ::std::optional<at::Tensor> &found_inf = {})