torch.nn.utils.clip_grad.clip_grad_value_

torch.nn.utils.clip_grad.clip_grad_value_(parameters, clip_value, foreach=None) [исходный код]

Ограничивает градиенты итерируемого объекта параметров заданным значением.

Градиенты изменяются на месте.

Параметры:

• parameters (Iterable[Tensor] or Tensor) – итерируемый объект из тензоров или один тензор, градиенты которого будут нормализованы
• clip_value (float) – максимально допустимое значение градиентов. Градиенты ограничиваются диапазоном $\left[\text{-clip\_value}, \text{clip\_value}\right]$
• foreach (bool, optional) – использовать более быструю реализацию на основе foreach. Если None, использовать реализацию foreach для собственных тензоров CUDA и CPU и незаметно переключаться на медленную реализацию для других типов устройств. Значение по умолчанию: None