torch.nn.Unfold accepts 0-dim for batch size (#40689)
Summary:
In partial completion of https://github.com/pytorch/pytorch/issues/12013
Allows specifying a tensor with 0-dim batch size for `torch.nn.Unfold()`.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/40689
Reviewed By: zou3519
Differential Revision: D24441164
Pulled By: ngimel
fbshipit-source-id: 49cd53b9b23f2e221aecdb4b5fed19a234038063