torch.nn.utils.clip_grad_norm

torch.nn.utils.clip_grad_norm(parameters, max_norm, norm_type=2.0, error_if_nonfinite=False, foreach=None)

對可迭代引數的梯度範數進行裁剪。

警告

此方法現已棄用，推薦使用 torch.nn.utils.clip_grad_norm_()。

返回型別

張量