1,224,224] to have 3 channels,根源是 batch 维度放置位置错误">报错 expected input [3,1,224,224] to have 3 channels,根源是 batch 维度放置位置错误

出现 expected input [3,1,224,224] to have 3 channels 这个 PyTorch 图像输入报错,核心原因就是张量维度顺序搞错、batch 维