




















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
此内容由惯性聚合(RSS阅读器)自动聚合整理,仅供阅读参考。 原文来自 — 版权归原作者所有。