Solution 23

Resize_ function in Pytorch distorts the image