fix wrong schema (#19370)
Summary:
Op was improperly schematized previously. Evidently checkScript does not test if the outputs are the same type.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/19370
Differential Revision: D14985159
Pulled By: eellison
fbshipit-source-id: feb60552afa2a6956d71f64801f15e5fe19c3a91
