I have an image and I want to pass it through convolution net and the output of the net is the same dimensional matrix as the input.
eg. Suppose I have an input of 3x32x32 the output is the same as 32x32.
Now I want to calculate the loss with a label of 32x32 but pytorch does not support multi-dimensional label to be fed in the loss function.
Is there a way to do this