I guess it is becuase default shape of the image is should be like image size, image size, image size,  suppose like 128,128,128. I am taking 3D images where no of patches of the fixed image and moving image are not same. So I am trying to make this model take input of 1,1, any value, 128, 128.