support discontiguous tensors only for contiguous output format (#57177)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/57122
Pull Request resolved: https://github.com/pytorch/pytorch/pull/57177
Reviewed By: zou3519
Differential Revision: D28072674
Pulled By: ngimel
fbshipit-source-id: 1f0b1d6916eb9739c35a5ac5aba33e70c1c43a34
