torch.nn.utils.clip_grad_value_

torch.nn.utils.clip_grad_value_(parameters, clip_value, foreach=None) [source]

Ограничивает градиент итеративного набора параметров до заданного значения.

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

Parameters

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