Port NumPy typing testing style to PyTorch (#52408)
Summary:
ref: https://github.com/pytorch/pytorch/issues/16574
Pull Request resolved: https://github.com/pytorch/pytorch/pull/52408
Reviewed By: anjali411
Differential Revision: D26654687
Pulled By: malfet
fbshipit-source-id: 6feb603d8fb03c2ba2a01468bfde1a9901e889fd
