Ah OK, so it does come from PyTorch. Great, I was wondering how it was working.
Also thank you for that bit of info about the colormap. I actually like the way it looks.
Thanks for always have a good answer!