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/算法/框架的路上作伴

相关文章

评论可见，请评论后查看内容，谢谢！！！
 您阅读本篇文章共花了：