pytorch 报错inf nan

1：a中含有inf值，替换所有inf为1或者255

a = torch.where(torch.isinf(a), torch.full_like(a, 1), a)
其中1-255范围内对于log(1+exp)的值没太大影响

2：含nan则

a = torch.where(torch.isnan(a), torch.full_like(a, 1), a)

3：where函数

torch.where(a>0,a,b) # 满足a>0的位置处，合并a和b

torch.where()函数解读_wang xiang的博客-CSDN博客_torch where

4：full_like函数

torch.full_like(a,1) #创建维度与a相同，元素值全为1的

pytorch每日一学23(torch.full()、torch.full_like())创造全value的矩阵_Fluid_ray的博客-CSDN博客