Fix formatting issues in torch.tensor_split documentation (#46328)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/46328
Reviewed By: heitorschueroff
Differential Revision: D24318003
Pulled By: mruberry
fbshipit-source-id: 140d391dd927ff3374dd6c4c6e2da7cb67417b31