pytorch

e5c2a35d - Add check that embedding_bag's weight is 2D (#94931)

Add check that embedding_bag's weight is 2D (#94931) Fixes https://github.com/pytorch/pytorch/issues/94445 Pull Request resolved: https://github.com/pytorch/pytorch/pull/94931 Approved by: https://github.com/albanD