torch.nn.utils.clip_grad.clip_grad_norm

torch.nn.utils.clip_grad.clip_grad_norm(parameters, max_norm, norm_type=2.0, error_if_nonfinite=False, foreach=None) [исходный код]

Ограничивает норму градиента для итерируемой последовательности параметров.

Предупреждение

Этот метод устарел; вместо него используйте torch.nn.utils.clip_grad_norm_().

Тип возвращаемого значения:

Tensor