评价此页

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