評價此頁

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) – 將要被歸一化梯度的 Tensor 可迭代物件或單個 Tensor

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

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