How to convert a matrix to heatmap image in torch

Thank u, i find this is very useful