Removed unecessary check in `select_nested` (#89150)
Implementation in #88585 should work for all dimensions. Removed unnecessary check that constrained select to dims 0 and 1
Pull Request resolved: https://github.com/pytorch/pytorch/pull/89150
Approved by: https://github.com/cpuhrsch