Add prim, ref, and OpInfo for arange (#81734)
Per title.
- See https://github.com/pytorch/pytorch/issues/81959 for discussion on overloading
TODO:
- ~Handle remaining TensorOptions: layout, pin_memory (won't do in this PR)~
- ~Add sample inputs for floating point and complex numbers (done for floating point, won't do for complex in this PR)~
Pull Request resolved: https://github.com/pytorch/pytorch/pull/81734
Approved by: https://github.com/ngimel