torch.Tensor.nan_to_num

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

See torch.nan_to_num().