[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
