Hi, thank you always for your help.
I am trying to convert an image set stored in torch.Tensor of like [number_of_images, width, height],
to a set of color images of like [number_of_images, 3, width, height].
I would like to use the color scheme ‘magma’ that can be used in matplotlib.
Is there any good way to realize it?
Thank you in advance