pytorch

d301fb40 - Fix broken doc tests after #108482 (#108725)

Fix broken doc tests after #108482 (#108725) Tiny fix so I don't wanna revert the PR. Pull Request resolved: https://github.com/pytorch/pytorch/pull/108725 Approved by: https://github.com/kit1980