pytorch

66cf1b64 - correct argument name in docs (#81485)

correct argument name in docs (#81485) Recently introduced `average_attn_weights` argument is documented incorrectly. Pull Request resolved: https://github.com/pytorch/pytorch/pull/81485 Approved by: https://github.com/albanD