RuntimeError: Sizes of tensors must match except in dimension 1. Expected size 1 but got size 2 for tensor number 1 in the list

这个错误通常是因为你在对不同形状的张量进行操作时出错了。在你的代码中，有一个张量的大小与其他张量的大小不匹配，尤其是在第一个维度上，因为错误信息中提到的是大小1和2。请检查你的代码，确保所有涉及大小的操作都是正确的。你可以使用 PyTorch 的 `size()` 函数来查看张量的大小，以帮助你找到问题。