У меня есть тензор, размер которого (batch_size, seqLength, seqLength, label_number), и тензор маски с размером (batch_size, seqLength), когда я вычисляю потери, мне нужно замаскировать слово, которое я дополняю. Мой Вопрос в том, как использовать эту 2-мерную маску для маскировки образцов заполнения в 4-мерном тензоре, чтобы я мог рассчитать правильные потери?