Move comment back to the class it describes (it accidentally drifted away).
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/27148
Test Plan: Imported from OSS
Differential Revision: D17691005
Pulled By: ZolotukhinM
fbshipit-source-id: ac2aa5c8706e4a87ccc3a6d6a9baa3d6c8fb6e82
Author
Mikhail Zolotukhin