The size of tensor a (50) must match the size of tensor b (10) at non-singleton dimension 1怎么修改
时间: 2023-09-24 10:13:35 浏览: 158
Pytorch 之修改Tensor部分值方式
5星 · 资源好评率100%
要解决这个问题,你需要将这两个 tensor 在第一维的大小进行调整,使它们的大小相同。具体的方法取决于你的代码实现,以下是一些可能的解决方案:
1. 在 tensor a 上使用 slice 操作,将第一维的大小从 50 调整为 10。
2. 在 tensor b 上使用 repeat 操作,将第一维的大小从 10 调整为 50。
3. 在 tensor a 和 tensor b 上同时使用 reshape 操作,将它们的第一维大小都调整为一个相同的数值。具体的调整方式和参数设置根据代码实现而定。
总之,你需要通过某种方式将这两个 tensor 在第一维的大小调整为相同,才能解决这个错误。
阅读全文