【Pytorch】torchvision.utils.make

【Pytorch】torchvision.utils.make

其官方文档给出的解释如下：

该方法可以用来直接将数据结构[batch,channel.height,width]形式的图像转化为图像矩阵，便于将多张图像进行可视化。其具体的应用可以参考我的另一篇博客《【Pytorch】利用Pytorch实现基于线性层的自编码模型》。其原理如下，在该篇博客中，b_x为一个[60000，784]的张量，经过reshape处理后，将其分割为60000个每一个都是28*28的矩阵，由此则可以按照图像格式进行拼凑后再进行输出。其展示如下：