Shortcuts

Function at::_amp_foreach_non_finite_check_and_unscale

Function Documentation

inline ::std::tuple<::std::vector<at::Tensor>, at::Tensor> at::_amp_foreach_non_finite_check_and_unscale(at::TensorList self, const at::Tensor &found_inf, const at::Tensor &inv_scale)

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources