Well yeah because i have another input and output size of my image. Or do i get something wrong?
Because for input i have 10 batches a 1 channel(gray iamge) and then 560x656.
And as output i need a vector of x,y,click