math module support (#19115)
Summary:
This PR refer to issue [#19026](https://github.com/pytorch/pytorch/issues/19026)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/19115
Differential Revision: D14936053
Pulled By: driazati
fbshipit-source-id: 68d5f33ced085fcb8c10ff953bc7e99df055eccc
