The size of tensor a (4) must match the size of tensor b (2) at non-singleton dimension 1

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() 展开即可解决问题

你可能感兴趣的:(bug,pytorch)