PyTorch RuntimeError: 张量维度不匹配 (total_grad.add_(params.grad))

这个错误是由于两个张量的维度不匹配导致的。在这种情况下，张量 'a' 的第三维的大小为 7，而张量 'b' 的第三维的大小为 64。要解决这个问题，需要确保这两个张量在所有维度上的大小匹配。可能需要检查代码中的张量维度和大小是否正确，并进行相应的更改。

原文地址: https://www.cveoy.top/t/topic/nw1k 著作权归作者所有。请勿转载和采集!