在pytorch进行一些计算的时候,当某个数据需要出现在分母或者对数函数中的时候,我们需要保证其不是零,从而保证运算的稳定性,可以使用下面的代码来实现:
x = torch.tensor([0.2,0.1,0,0.4])
print(x)
x.clamp_(1e-6)
print(x)
上述程序的运行结果为:
tensor([0.2000, 0.1000, 0.0000, 0.4000])
tensor([2.0000e-01, 1.0000e-01, 1.0000e-06, 4.0000e-01])
使用这个语句,保证了张量中最小的数值为1e-6,从而避免了结果为零的情况