The size of tensor a (123) must match the size of tensor b (256) at non-singleton dimension 1
时间: 2023-12-13 12:29:58 浏览: 160
tensor:包张量在Go中提供了有效且通用的n维数组,可用于机器学习和深度学习
这个错误提示是 PyTorch 给出的,它表示在进行某个操作时,两个 tensor 的形状不匹配。具体来说,tensor a 在第 1 维的大小是 123,而 tensor b 在第 1 维的大小是 256,这两个大小不一致,因此无法进行操作。
解决这个问题的方法是调整 tensor 的形状,使得它们在需要匹配的维度上的大小一致。具体的调整方法要根据具体的操作和数据形状来确定。
阅读全文