Pytorch bilinear interpolation. view(1, 512, -1) input2_flat = input2.