Accept 0-dim channel inputs in convolution layer (#66256)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/56998 .
Pull Request resolved: https://github.com/pytorch/pytorch/pull/66256
Reviewed By: mrshenli
Differential Revision: D31859428
Pulled By: jbschlosser
fbshipit-source-id: 034b6c1ce35aac50eabfa09bbcd8b1e3c8b171bd