本文主要是介绍NPU float(“inf“) mask_fill 出现NAN,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
使用NPU时,采用mask_fill函数,会出错:
tensors.masked_fill(mask.unsqueeze(-1), float('inf'))
无法直接使用表达式,会报错 NAN:
取一个较大的值替换即可;
tensors.masked_fill(mask.unsqueeze(-1), float(1e10))
这篇关于NPU float(“inf“) mask_fill 出现NAN的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!