The size of tensor a (4) must match the size of tensor b

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

1. 调试查看两个张量的维度
2. 将其修改成一致的

我主要问题是其中一个多了一维, 用pytorch 的squeeze() 展开即可解决问题