migrate cuda implementation of take() from TH to ATen (#45430)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/45430
Test Plan: Imported from OSS
Reviewed By: mrshenli
Differential Revision: D24037297
Pulled By: bdhirsh
fbshipit-source-id: 7c5f2c08e895fb0c25eec1d68c7455e4f2b1c64e
