Does nn.PixelShuffle support 3D data?

Does nn.PixelShuffle support 3D data? In other words, the input is 5D.

Based on the docs a 4-dimensional input tensor is expected and you should also get a RuntimeError, if your input tensor has a different number of dimensions.

is there any solution to 3d data in nn.PixelShuffle or will be in the future?

This PR might have already added the support.
You could use the latest nightly binary or build from source and check, if it’s working. :slight_smile:

The latest nightly binary solved the problem of nn.PixelShuffle for input with more than 4 dimensions.