stack expects each tensor to be equal size, but got [2, 256] at entry 0 and [2, 128] at entry 79

这个错误可能是由于你使用了`torch.stack`函数将不同形状的张量堆叠在一起，导致了张量的形状不匹配。

你需要确保每个张量的形状相同才能使用`torch.stack`函数。你可以使用`torch.cat`函数将不同形状的张量串联在一起，而不是堆叠它们。或者你可以在堆叠之前通过重塑或填充操作使它们具有相同的形状。

例如，如果你有一个形状为`[2, 256]`的张量和一个形状为`[2, 128]`的张量，你可以使用以下代码将它们串联在一起：

tensor1 = torch.randn(2, 256)
tensor2 = torch.randn(2, 128)

# 串联张量
concatenated_tensor = torch.cat([tensor1, tensor2], dim=1)

print(concatenated_tensor.shape) # 输出：torch.Size([2, 384])

这将创建一个形状为`[2, 384]`的张量，其中第二个维度是`256 + 128 = 384`。

希望这可以帮助你解决问题！