FIX Validates that weights are 2d in embedding (#59314)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/55185
Pull Request resolved: https://github.com/pytorch/pytorch/pull/59314
Reviewed By: H-Huang
Differential Revision: D28837753
Pulled By: jbschlosser
fbshipit-source-id: 683378244c61b0937c95563f91ef87ab09fd1653
