[wip] doc_fix (#51825)
Summary:
tries to fix doc_test
Pull Request resolved: https://github.com/pytorch/pytorch/pull/51825
Reviewed By: bertmaher
Differential Revision: D26295583
Pulled By: ngimel
fbshipit-source-id: 13f6e7f1675d810adfd4abd2d579e2812fe54c80
