Add python test for extension backend tensor.device (#17602)
Summary:
Adding a test for #17361
Pull Request resolved: https://github.com/pytorch/pytorch/pull/17602
Differential Revision: D14287373
Pulled By: li-roy
fbshipit-source-id: 544ecf17eb310aed22ba0ea5f86f46b8e3bb69b5
