pytorch

533f0cb2 - Set correct module for APIs in torch module

Set correct module for APIs in torch module Pull Request resolved: https://github.com/pytorch/pytorch/pull/75801 Approved by: https://github.com/albanD