torch.Tensor.nan_to_num

Tensor.nan_to_num(nan=0.0, posinf=None, neginf=None) → Tensor

参见 torch.nan_to_num()。