The size of tensor a (4) must match the size of tensor b (2) at non-singleton dimension 1 意思就是 a 张量和b张量 维度不一致, 无法完成操作 non-singleton dimension 1 在非单一维度上 维度不一致 solution 调试查看两个张量的维度将其修改成一致的 我主要问题是其中一个多了一维, 用pytorch 的squeeze() 展开即可解决问题