标签： pytorch
我有一个张量，大小为（batch_size，seqLength，seqLength，label_number），还有一个掩码张量，其大小为（batch_size，seqLength），在计算损失时，我需要掩盖填充的单词。我问题是如何使用此2d遮罩遮盖4d张量中的填充样本，以便我可以计算正确的损耗？