[tp] ufmt test/distributed/tensor (#89970)
formatting stack to make dtensor and tp align with pytorch format standard.
cmd: `ufmt format test/distributed/tensor`
Pull Request resolved: https://github.com/pytorch/pytorch/pull/89970
Approved by: https://github.com/fduwjj
