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_().

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

Тензор