快捷方式

torch.nn.utils.get_total_norm

torch.nn.utils.get_total_norm(tensors, norm_type=2.0, error_if_nonfinite=False, foreach=None)[source]

計算可迭代張量的範數。

範數是針對單個張量的範數計算的,就像單個張量的範數被連線成一個向量一樣。

引數
  • tensors (Iterable[Tensor] or Tensor) – 可迭代的張量或將被歸一化的單個張量

  • norm_type (float) – 使用的 p-範數型別。對於無窮範數可以是 'inf'

  • error_if_nonfinite (bool) – 如果為 True,則當 tensors 的總範數是 naninf-inf 時丟擲錯誤。預設值: False

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

返回

張量(視為單個向量)的總範數。

返回型別

Tensor

文件

訪問 PyTorch 全面開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深度教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源