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 的總範數是 nan、inf 或 -inf 時丟擲錯誤。預設值: False foreach (bool) – 使用更快的基於 foreach 的實現。如果為 None,則對 CUDA 和 CPU 原生張量使用 foreach 實現,對其他裝置型別則靜默回退到慢速實現。預設值: None 返回 張量(視為單個向量)的總範數。 返回型別 Tensor