Add types argument to __torch_function__ (#34303)
Summary:
This PR adds the `types` argument to `__torch_function__` as per RFC 0001: https://github.com/pytorch/rfcs/pull/3
Pull Request resolved: https://github.com/pytorch/pytorch/pull/34303
Differential Revision: D20474992
Pulled By: ezyang
fbshipit-source-id: cdd40b3b38f3bda4ece8812a629f5db87e919d01
