[nn] lstm : no batch dim support (#71056)
Summary:
Reference: https://github.com/pytorch/pytorch/issues/60585
TODO:
* [x] Update docs
Pull Request resolved: https://github.com/pytorch/pytorch/pull/71056
Reviewed By: samdow
Differential Revision: D33638643
Pulled By: jbschlosser
fbshipit-source-id: c0949829de8a8e6e7b2873f459a8d7da597a3be3
(cherry picked from commit f94d5849f66dd7da2ae4037b7c1d3e72817e926f)