weixin_45078217的博客

原创 关于np.transpose()的理解

网上关于np.transpose()的函数解释有很多，但没找到有对实际意义的解释。我的理解是:在pytorch中张量tensor对图象的存储是(b,c,w,h)分别表示(图片数量，通道数，图片高，图片宽)。单独说tensor中的某张图片，也就是(管道数，宽，高)。而标准的rbg图象是(宽，高，管道数)。所以在pytorch中为了显示图象 需要通过np.transpose()对图象进行处理，例子:在这里贴一段有关处理minist数据集的代码片，大家只需要关注输出时np.transpose()前后sha