pytorch
44c9185f - Fix empty input issue of convolution for channels last memory format (#86521)

Commit
2 years ago
Fix empty input issue of convolution for channels last memory format (#86521) Fixes empty input convolution issue : when input is empty e.g. shape of (0, 3, 3, 4) and weight is channels last format, at::_unsafe_view will raise "view size is not compatible with input tensor's size and stride (at least one dimension spans across two contiguous subspaces). Use .reshape(...) instead." Pull Request resolved: https://github.com/pytorch/pytorch/pull/86521 Approved by: https://github.com/jgong5, https://github.com/malfet
Author
Committer
Parents
Loading