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
好记性不如烂键盘---点滴、积累、进步!

浙公网安备 33010602011771号