报错 The size of tensor a (133) must match tensor b (10)分类任务改用 MSELoss 的标签维度排查方案

这个报错本质是张量维度不匹配,在分类任务把损失函数从 CrossEntropyLoss 换成 MSELoss 时非常容易踩坑:模型输出 tensor a 形状是