Is your feature request related to a problem? Please describe.
Modify the pixelshuffle method in such a way that when using this upsampling method on any network, it can be scripted using torch.jit.script
Describe the solution you'd like
This should not raise an error:
import torch
from monai.networks.utils import pixelshuffle
torch.jit.script(pixelshuffle)