快捷方式

torch.nn.utils.remove_weight_norm

torch.nn.utils.remove_weight_norm(module, name='weight')[source][source]

從模組中移除權重歸一化重引數化。

引數
  • module (Module) – 包含權重歸一化的模組

  • name (str, optional) – 權重引數的名稱

返回值型別

T_module

示例

>>> m = weight_norm(nn.Linear(20, 40))
>>> remove_weight_norm(m)

文件

查閱 PyTorch 全面的開發者文件

檢視文件

教程

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

檢視教程

資源

查詢開發資源並解答疑問

檢視資源