Fix output type of torch.max for Tensor subclasses. (#47110)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/47090
Pull Request resolved: https://github.com/pytorch/pytorch/pull/47110
Reviewed By: ngimel
Differential Revision: D24649568
Pulled By: ezyang
fbshipit-source-id: 9374cf0c562de78e520bcb03415db273c1dd76a3