AdaptiveLogSoftmaxWithLoss no_batch_dim support (#69054)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/69054
Test Plan: Imported from OSS
Reviewed By: jbschlosser
Differential Revision: D33200166
Pulled By: george-qi
fbshipit-source-id: 9d953744351a25f372418d2a64e8402356d1e9b7