Fix doc (#53996)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/53996
Fixes issue: #52479
Test Plan: Imported from OSS
Reviewed By: anjali411
Differential Revision: D27051056
Pulled By: nikithamalgifb
fbshipit-source-id: ff5d2fc3599571346e2323fa893c1e238097a164