![shape '[-1, 1, 28, 28]' is invalid for input of size 55125,大罗洞观 - debug,人工智能,python,pytorch,numpy,pandas](https://imgs.yssmx.com/Uploads/2024/01/784575-1.jpeg)
食用指南：
框架使用过程中的问题首先要注意版本发布时间造成方法弃用当你在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 - 错误场景：
![shape '[-1, 1, 28, 28]' is invalid for input of size 55125,大罗洞观 - debug,人工智能,python,pytorch,numpy,pandas](https://imgs.yssmx.com/Uploads/2024/01/784575-2.png)
解决方案：
问题原因：
- 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)
- 解决后：
![shape '[-1, 1, 28, 28]' is invalid for input of size 55125,大罗洞观 - debug,人工智能,python,pytorch,numpy,pandas](https://imgs.yssmx.com/Uploads/2024/01/784575-3.png)
安利时间：
-
如果本篇博客解决了你的问题，不妨留下你的👍
让博主知道原来也有小伙伴在同一个坑摔倒了啊hh文章来源：https://www.toymoban.com/news/detail-784575.html
-
欢迎关注大二的博主，在学习C/C++/JAVA/算法/框架的路上作伴文章来源地址https://www.toymoban.com/news/detail-784575.html
到了这里，关于PyTorch报错shape ‘[16, 1, 28, 28]‘ is invalid for input of size 6272?尝试设置-1的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！
