torch.nn.utils.clip_grad_value_

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

將引數的可迭代物件的梯度裁剪到指定值。

梯度會原地修改。

引數

• parameters (Iterable[Tensor] 或 Tensor) – 將被裁剪梯度的 Tensor 可迭代物件或單個 Tensor

• clip_value (float) – 梯度的最大允許值。梯度將被裁剪到範圍 $\left[\text{-clip\_value}, \text{clip\_value}\right]$ 內

• foreach (bool) – 使用更快的基於 foreach 的實現。如果為 None，則對 CUDA 和 CPU 原生張量使用 foreach 實現，並對其他裝置型別靜默回退到慢速實現。預設值：None