mixmo.utils.visualize.
make_image
Transform a CIFAR numpy image into a pytorch image (need to swap dimensions)