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