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

在自学小土堆pytorch的过程中，在transform的章节遇到一个问题

用自己上传的图片进行标准化时，上传的图片由PIL转为Tensor类型时为4维张量，与3维张量不匹配。

报错内容如下

分析原因

定位到错误行16行，里面的参数img_tensor

查看img_tensor的维度，发现其维度是4，与trans_norm中的3维不匹配。

解决问题

将

img = Image.open("image/image.png")

改为

with open("image/image.png",'rb') as f:
    img = Image.open(f).convert("RGB")

记录一下自己解决问题的过程，如有缺漏，欢迎大佬指正！！

参考文章

Pytorch报错解决：The size of tensor a (4) must match the size of tensor b (3) at non-singleton dimensio-CSDN博客