使用NPU时,采用mask_fill函数,会出错:
tensors.masked_fill(mask.unsqueeze(-1), float('inf'))
无法直接使用表达式,会报错 NAN:
取一个较大的值替换即可;
tensors.masked_fill(mask.unsqueeze(-1), float(1e10))
使用NPU时,采用mask_fill函数,会出错:
tensors.masked_fill(mask.unsqueeze(-1), float('inf'))
无法直接使用表达式,会报错 NAN:
取一个较大的值替换即可;
tensors.masked_fill(mask.unsqueeze(-1), float(1e10))