等同于torchnan_to_num的方法

PyTorch中没有直接等同于torch.nan_to_num的方法，但可以使用以下方法实现类似的功能：

import torch

def nan_to_num(x, num=0.0, nan=0.0, inf=0.0, neg_inf=0.0):
    x[x != x] = nan
    x[x == float('inf')] = inf
    x[x == float('-inf')] = neg_inf
    x[torch.isnan(x)] = nan
    return x

# Example usage:
x = torch.tensor([1.0, float('nan'), float('inf'), float('-inf')])
print(nan_to_num(x))
# Output: tensor([ 1.,  0.,  0.,  0.])

此方法将NaN替换为给定的nan值，正无穷替换为inf值，负无穷替换为neg_inf值，同时将所有非数值替换为给定的num值。