Add context for warning about batch_first (#106139)
Summary: Add context for warning about batch_first
Test Plan: sandcastle github
Differential Revision: D47809651
Pull Request resolved: https://github.com/pytorch/pytorch/pull/106139
Approved by: https://github.com/mikaylagawarecki