Update MHA doc string (#97046) (#99746)
Summary: Update MHA doc string
Test Plan: sandcastle & github
Differential Revision: D44179519
Pull Request resolved: https://github.com/pytorch/pytorch/pull/97046
Approved by: https://github.com/voznesenskym
Co-authored-by: Michael Gschwind <mikekg@meta.com>