[pytorch] Add support for "height" and "width" dimension for the "select" operator on pytorch vulkan backend (#94612)
Summary: Add support for "height" and "width" dimension for the "select" operator on pytorch vulkan backend.
Test Plan:
```
yipjustin@yipjustin-mbp fbsource % buck run -c pt.vulkan_full_precision=1 --target-platforms ovr_config//platform/macos:arm64-fbsource //xplat/caffe2:pt_vulkan_api_test_binAppleMac\#macosx-arm64 -- --gtest_filter="*select_3d*"
Downloaded 1/2 artifacts, 1.29 Mbytes, 0.0% cache miss (for updated rules)
Building: finished in 3.7 sec (100%) 450/450 jobs, 2/450 updated
Total time: 3.8 sec
BUILD SUCCEEDED
Running main() from xplat/third-party/gmock/googletest-1.12.1/googletest/src/gtest_main.cc
Note: Google Test filter = *select_3d*
[==========] Running 9 tests from 1 test suite.
[----------] Global test environment set-up.
[----------] 9 tests from VulkanAPITest
[ RUN ] VulkanAPITest.select_3d_depth_small
[ OK ] VulkanAPITest.select_3d_depth_small (30 ms)
[ RUN ] VulkanAPITest.select_3d_depth_medium
[ OK ] VulkanAPITest.select_3d_depth_medium (0 ms)
[ RUN ] VulkanAPITest.select_3d_depth_large
[ OK ] VulkanAPITest.select_3d_depth_large (1 ms)
[ RUN ] VulkanAPITest.select_3d_height_small
[ OK ] VulkanAPITest.select_3d_height_small (0 ms)
[ RUN ] VulkanAPITest.select_3d_height_medium
[ OK ] VulkanAPITest.select_3d_height_medium (0 ms)
[ RUN ] VulkanAPITest.select_3d_height_large
[ OK ] VulkanAPITest.select_3d_height_large (3 ms)
[ RUN ] VulkanAPITest.select_3d_width_small
[ OK ] VulkanAPITest.select_3d_width_small (0 ms)
[ RUN ] VulkanAPITest.select_3d_width_medium
[ OK ] VulkanAPITest.select_3d_width_medium (0 ms)
[ RUN ] VulkanAPITest.select_3d_width_large
[ OK ] VulkanAPITest.select_3d_width_large (1 ms)
[----------] 9 tests from VulkanAPITest (40 ms total)
[----------] Global test environment tear-down
[==========] 9 tests from 1 test suite ran. (40 ms total)
[ PASSED ] 9 tests.
```
Reviewed By: SS-JIA
Differential Revision: D43020796
Pull Request resolved: https://github.com/pytorch/pytorch/pull/94612
Approved by: https://github.com/SS-JIA