torch.nn.utils.prune.identity

torch.nn.utils.prune.identity(module, name)

應用剪枝重引數化，但不對任何單元進行實際剪枝。

透過以下方式，對 module 中名為 name 的張量應用剪枝重引數化，而不實際剪枝任何單元。修改 module（並返回修改後的 module）：

1. 新增一個名為 name+'_mask' 的命名緩衝區，對應於剪枝方法應用於引數 name 的二值掩碼。

2. 用剪枝後的版本替換引數 name，同時將原始（未剪枝）引數儲存在一個名為 name+'_orig' 的新引數中。

注意

掩碼是一個全為 1 的張量。

引數

• module (nn.Module) – 包含要剪枝張量的模組。

• name (str) – 在 module 中執行剪枝操作的引數名稱。

返回

模組的修改（即剪枝）後的版本

返回型別

module (nn.Module)

示例

>>> m = prune.identity(nn.Linear(2, 3), "bias")
>>> print(m.bias_mask)
tensor([1., 1., 1.])