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 nn.conv2d() ? Like nn.conv2d(in_channels=3, out_channels=64) ?