Thanks for the reply.
However what I really wanted is convolution between two images by using torch.nn.functional.conv2d
In other word, I am trying to use smaller image as convolution kernel.
Having real hard time here…