pytorch笔记： 处理inf和nan数值

Posted 2021-11-04 UQI-LIUWJ

import torch
import numpy as np
a = torch.Tensor(
    [[1, 2, np.nan], 
    [np.inf, np.nan, 4], 
    [3, 4, 5]])
'''
tensor([[1., 2., nan],
        [inf, nan, 4.],
        [3., 4., 5.]])
'''

假设我们有这样的一个pytorch tensor，我们需要把其中的inf和nan填充成别的数字

1 把nan值设置为0

a1 = torch.where(
        torch.isnan(a), 
        torch.full_like(a, 0), 
        a)
#where的第一行为条件限制，如果满足条件，则选择下一个参数，否则选择下下个参数作为输出。

'''
(tensor([[1., 2., 0.],
         [inf, 0., 4.],
         [3., 4., 5.]]),
 tensor([[1., 2., nan],
         [inf, nan, 4.],
         [3., 4., 5.]]))
'''

2 把inf值设置为0

a2 = torch.where(
        torch.isinf(a), 
        torch.full_like(a, 0), 
        a)
 
a2,a
'''
(tensor([[1., 2., nan],
         [0., nan, 4.],
         [3., 4., 5.]]),
 tensor([[1., 2., nan],
         [inf, nan, 4.],
         [3., 4., 5.]]))
'''