docs: fix parameter in `torch.take` (#54667)
Summary:
fixes https://github.com/pytorch/pytorch/issues/43495
https://11812612-65600975-gh.circle-artifacts.com/0/docs/generated/torch.take.html
Pull Request resolved: https://github.com/pytorch/pytorch/pull/54667
Reviewed By: ailzhang
Differential Revision: D27328252
Pulled By: zou3519
fbshipit-source-id: 5812ebdaba063ca0a9c0f4a9becd00a570d84d30
