My input tensor size is (72, 72) and the channel is 3. I have used
nn.conv2d() in my network. And I want to use
nn.PixelShuffle() after the convolution layers to get a HR ouput. I have learned that
nn.PixelShuffle() gets a input of (4, 72, 72) and give an output in (1, 144,144). I have several issues.
I think the tensor dimension cannot be expanded during training. So is my input of my network supposed to be in (3, 4, 72, 72)?
If so, how to set the channel parameters in