pytorch

01e92f2a - [nn] no batch dim support: CosineEmbeddingLoss (#64590)

[nn] no batch dim support: CosineEmbeddingLoss (#64590) Summary: Reference: https://github.com/pytorch/pytorch/issues/60585 TODO * [x] Add tests Pull Request resolved: https://github.com/pytorch/pytorch/pull/64590 Reviewed By: H-Huang Differential Revision: D30900775 Pulled By: jbschlosser fbshipit-source-id: d24e72787017e79afbf8f04a94901a290485b81a