pytorch

04224e18 - Docs: Detail 3D tensor shape for transformer masks

Docs: Detail 3D tensor shape for transformer masks Fixes #74612. Pull Request resolved: https://github.com/pytorch/pytorch/pull/75552 Approved by: https://github.com/albanD