[nn] TransformerEncoderLayer: no-batch-dim (#69291)
Summary:
Reference: https://github.com/pytorch/pytorch/issues/60585
TODO:
* [ ] Update docs?
* [x] Generic reference function?
cc albanD mruberry jbschlosser walterddr kshitij12345
Pull Request resolved: https://github.com/pytorch/pytorch/pull/69291
Reviewed By: davidberard98
Differential Revision: D33278970
Pulled By: jbschlosser
fbshipit-source-id: 8dd5b6d7c0099fa38aa037c186778b10834bdee4