这几天写代码,一到展示图片就头疼。有的读进来是 C H W ,有的是 H W C ,tensor 要送 numpy 和 cpu ,如果在 0 到 1 还要手动乘 255 。很多时候我只是想随手看一眼图片而已,要求不高,有没有造好的轮子可以通用地展示图片,不去操心 shape 之类的事情? 图片, 展示, 通用, tensor