@InnovArul  @justusschock
I deactivated it, but it is still not giving the answer it should…
Update:
It is a little wierd…
When i do
Input = torch.rand(1,1,6,6)
it works
but when i do:
Input = torch.rand(1,2,6,6)
it doesnot