The size of tensor a (3) must match the size of tensor b (512) at non-singleton dimension 2这个错误什么意思
时间: 2023-11-06 07:42:28 浏览: 11
double 计算过程出现的误差
5星 · 资源好评率100%
这个错误意味着你正在尝试执行一个操作,其中两个张量的形状不兼容。具体来说,它表示尝试使用一个形状为 (3) 的张量 a 和一个形状为 (其他维度..., 512) 的张量 b 进行某个操作,其中 a 的最后一个维度大小为 3,而 b 的第二个维度大小为 512,这不兼容。
可能的解决方案包括:重新设计你的操作,使得 a 和 b 的形状兼容;或者调整张量 a 的形状,使其在最后一个维度上的大小为 512,以便与张量 b 的形状匹配。
阅读全文