pytorch

bf04c2ca - Make torch checks same for both CPU and CUDA multinomial (#25595)

Make torch checks same for both CPU and CUDA multinomial (#25595) Summary: Currently we have different checks for multinomial method on CPU and CUDA. This PR will make them consistent. Pull Request resolved: https://github.com/pytorch/pytorch/pull/25595 Differential Revision: D17236163 Pulled By: ifedan fbshipit-source-id: 7718173bdaf216e8eb636c2a5b9c5939b975325b