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.]]))
'''