[PyTorch] Remove device check from a few indexing methods (#58800)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/58800
These methods leverages TensorIterator which will handle
(or skip) device check.
ghstack-source-id: 129654358
Test Plan: CI && sandcastle
Reviewed By: ngimel
Differential Revision: D28622626
fbshipit-source-id: 6153299780d4f7bf286423520ba4cb60b554335e