pytorch

2af64ba3 - Allow output to zero-strided tensors if the size is <= 1 along that dim (#34100)

Allow output to zero-strided tensors if the size is <= 1 along that dim (#34100) Summary: Fixes https://github.com/pytorch/pytorch/issues/33812 Pull Request resolved: https://github.com/pytorch/pytorch/pull/34100 Differential Revision: D20267778 Pulled By: ngimel fbshipit-source-id: 1b84c4f6e6bf5d29c3698daa3cb71554b25c1eee