torch.Tensor.nan_to_num

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

请参阅 torch.nan_to_num()。