RuntimeError: Has not supported replacing the module: `PixelShuffle`

Describe the issue:

Environment:

• NNI version:Release 2.3 (github master branch)
• Training service (local|remote|pai|aml|etc):local
• Client OS:Ubuntu 18.04
• Server OS (for remote mode only):
• Python version: Python 3.7.0
• PyTorch/TensorFlow version: 1.7.1
• Is conda/virtualenv/venv used?: virutalenv
• Is running in Docker?: no

For super resolution models, nn.PixelShuffle(scale) is widely used. When I tried to compress and speed up a super resolution model with nn.PixelShuffle( ), an error was met: RuntimeError: Has not supported replacing the module: PixelShuffle . Is there a way that I can fix or work around it. Thanks very much!