torch数据中的nan用0替代

可以使用以下代码将torch数据中的NaN替换为0：

import torch
import numpy as np

# create a sample tensor with NaN values
t = torch.tensor([[1, 2, 3], [4, np.nan, 6], [7, 8, np.nan]])

# replace NaN with 0
t[torch.isnan(t)] = 0

print(t)

输出结果为：

tensor([[1., 2., 3.],
        [4., 0., 6.],
        [7., 8., 0.]])