fm = torch.bmm(reduce_x, torch.softmax(attention, dim=1)) RuntimeError: batch1 dim 2 must match batch2 dim 1

这个错误通常是由于输入的矩阵维度不匹配导致的。具体来说，这里的错误是由于 `reduce_x` 的第二个维度与 `attention` 的第一个维度不匹配，导致无法进行矩阵乘法。

你需要检查一下 `reduce_x` 和 `attention` 的大小，并确保它们的维度能够匹配。你可能需要使用 `torch.transpose()` 或 `torch.unsqueeze()` 函数来调整矩阵的维度以匹配它们的形状。