f-dangel / unfoldNd / 7478206151

100%

main: 100%

push

github

[FIX] Treat padding explicitly when scattering, then unpad

This should fix #30.

The problem was in the creation of the tensor that maps the patch values
back to the positions in the original image. The code was using `unfold`
with non-zero padding, which would lead to a lot of patch values being
assigned to location 0. The fix is to deal with padding explicitly so
that the `unfold` operation uses `padding=0`. Then the index tensor is
correct, and after scattering we have to un-pad the image to obtain the
output.

12 of 12 new or added lines in 1 file covered. (100.0%)

150 of 150 relevant lines covered (100.0%)

5.96 hits per line