Fix logit doc (#41384)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/41384
Fix logit doc
Test Plan: unittest
Reviewed By: houseroad
Differential Revision: D22521730
fbshipit-source-id: 270462008c6ac73cd90aecd77c5de112fc93ea8d
