pytorch

2c556a94 - fix the input/output type mismatch (#20829)

fix the input/output type mismatch (#20829) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/20829 as title Reviewed By: jamesr66a Differential Revision: D15461937 fbshipit-source-id: 02c7150c0e8d020030ae8898008f718c74850dca