I’m a bit confused about the error message, as F.pixel_shuffle doesn’t expect a dimension argument, so it might be crashing internally?
Could you create an issue on GitHub so that we could track this issue?