pytorch

5737c525 - Fix for 16939:multinomial performance regressed

Fix for 16939:multinomial performance regressed Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/17121 Differential Revision: D14088558 Pulled By: ifedan fbshipit-source-id: e03583135f1e797fe1d8081ec5e9e6b63d4015c1