Does Pytorch have a function to decode a PNG image to a uint8 or uint16 tensor? I’m trying to track confusion matrix in tensorboard, and I found a way in Tensorflow to that converts the matplotlib plot specified by ‘figure’ to a PNG image, wanted to see if there was an equivalent for Pytorch.