specify data type in the doc (#19959)
Summary:
addresses comments in #19915
Pull Request resolved: https://github.com/pytorch/pytorch/pull/19959
Differential Revision: D15149993
Pulled By: orionr
fbshipit-source-id: 0e438cfa1a311e89d4bed7ae9d7710a9f1b19a78