pytorch

de67b52a - Unify `multi_margin_loss_shape_check` on CPU and CUDA (#104851)

Unify `multi_margin_loss_shape_check` on CPU and CUDA (#104851) Pull Request resolved: https://github.com/pytorch/pytorch/pull/104851 Approved by: https://github.com/ezyang