这个错误通常是因为您在使用概率分布时没有正确匹配输入和输出 shape 。 在 PyTorch 中，logits 是指输出的未进行 softmax 处理的概率分布，而 labels 是指实际的标签。

在您的代码中，logits 的维度是 [56, f = labels_dims[i]:252],而 labels 的维度是 [56, f = labels_dims[i]:252]。这两个维度不一致，因此会报错。请检查您的代码中是否正确计算了 logits 和 labels 的 shape。