nn.PixelShuffle expects an input in the shape [*, C * r**2, H, W] and outputs [*, C, H*r, W*r] as described in the docs.
If you want to use an input of [*, 4, 72, 72], r would correspond to 2 and thus C to 1.
In that case your conv layer should probably output 4 channels.