Change torch::Tensor to at::Tensor to fix build failure (#62425)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/62425
Fixes https://github.com/pytorch/pytorch/issues/62416
Test Plan: Imported from OSS
Reviewed By: suo
Differential Revision: D30000948
Pulled By: heitorschueroff
fbshipit-source-id: 07dfc88a01b7718bc32be4342f43bb2cf2842b60