PyTorch报错shape ‘[16, 1, 28, 28]‘ is invalid for input of size 6272?尝试设置-1

debug

• • 食用指南：
  • 框架/语言：
  • 问题描述：
  • 解决方案：
  • • 问题原因：
    • 解决方案：
    • 安利时间：

      食用指南：

      框架使用过程中的问题首先要注意版本发布时间造成方法弃用

      当你在CSDN等网站查找不到最新版本框架的问题解决，别迟疑，马上去Goole

      语言问题比框架问题好解决

      如果本篇debug博客解决了你的问题，不妨留下你的👍

      欢迎关注博主，查看更多c/c++/JAVA/算法/框架知识

      框架/语言：

      • PyTorch

        问题描述：

        • 使用tensor.reshape() 或 torch.reshape(data, (dim1, dim2, …)) 规范矩阵类型时，报错：

          shape ‘[16, 1, 28, 28]’ is invalid for input of size 6272

        • 错误场景：

          

          解决方案：

          问题原因：

          • reshape参数过于详细：

            默认每次加载的矩阵都是16*1*28*28

            data = data.reshape(16, 1, 28, 28)
            # 第一个参数是batch_size
            # 第二个参数的通道数目，灰度图一般为1，彩色图为RGB三通道
            # 之后的参数是图像的高度和宽度（可能还有深度）
            

            • dataloader可能有余数：

              batch_size表示每次取和同时处理的图像张数，但是可能图像总数并非正好是batch_size倍数

              解决方案：

              • reshape(-1, …)：

                最后一次迭代dataloader时，可能不足batch_size张图片，

                这时参数设为-1，表示有多少取多少即可

                data = data.reshape(16, 1, 28, 28)
                

                • 解决后：

                  

                  安利时间：

                  • 如果本篇博客解决了你的问题，不妨留下你的👍

                    让博主知道原来也有小伙伴在同一个坑摔倒了啊hh

                  • 欢迎关注大二的博主，在学习C/C++/JAVA/算法/框架的路上作伴