Op info test for `stft .. take_along_dim`

[qihqi]

Fix the Op info test for stft .. take_along_dim

1. Find the lines 318 to 322 of test_ops.py and remove stft .. take_along_dim from skip_list
2. Run op_info test with pytest test/test_ops.py
3. Fix the failure.

Please refer to this guide as guide to fix:

Also refer to these PRs:

• https://github.com/pytorch/xla/pull/7343
• https://github.com/pytorch/xla/pull/7312
• https://github.com/pytorch/xla/pull/7325