qq_43908211的博客

原创 The size of tensor a (4) must match the size of tensor b (3) at non-singleton dimension 0解决方法

用自己上传的图片进行标准化时，上传的图片由PIL转为Tensor类型时为4维张量，与3维张量不匹配。查看img_tensor的维度，发现其维度是4，与trans_norm中的3维不匹配。在自学小土堆pytorch的过程中，在transform的章节遇到一个问题。记录一下自己解决问题的过程，如有缺漏，欢迎大佬指正！定位到错误行16行，里面的参数img_tensor。