Shortcuts

torch.Tensor.nan_to_num_

Tensor.nan_to_num_(nan=0.0, posinf=None, neginf=None) 张量

原地版本的 nan_to_num()

优云智算