value = value.masked_fill(input_padding_mask[..., None], float(0))

value[b, 23052, 256], input_padding_mask[b, 23052]
input_padding_mask[..., None][b, 23052, 1] input_padding_mask.dtype torch.bool
value = value.masked_fill(input_padding_mask[..., None], float(0))

等价于:
value[input_padding_mask] = 0

posted @ 2026-03-16 15:10  无左无右  阅读(14)  评论(0)    收藏  举报