pytorch

e6a31545 - Allow broadcasting along non-reduction dimension for cosine similarity (#62912)

Allow broadcasting along non-reduction dimension for cosine similarity (#62912) Summary: Checks introduced by https://github.com/pytorch/pytorch/issues/58559 are too strict and disable correctly working cases that people were relying on. Pull Request resolved: https://github.com/pytorch/pytorch/pull/62912 Reviewed By: jbschlosser Differential Revision: D30165827 Pulled By: ngimel fbshipit-source-id: f9229a9fc70142fe08a42fbf2d18dae12f679646