torch.Tensor.nan_to_num

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

见 torch.nan_to_num() 。