torch.nn.utils.prune.identity¶
- torch.nn.utils.prune.identity(module, name)[源代码]¶
应用修剪重新参数化而不修剪任何单元。
将修剪重新参数化应用于对应于
module
中名为name
的参数的张量,而不实际修剪任何单元。通过以下方式修改模块以供就地使用(也返回修改后的模块):添加一个名为
name+'_mask'
的命名缓冲区,对应于由修剪方法应用于参数name
的二元掩码。用修剪后的版本替换参数
name
,同时将原始(未修剪的)参数存储在一个名为name+'_orig'
的新参数中。
注意
掩码是全为一的张量。
- 参数
- 返回值
输入模块的修改版(即修剪后的版本)
- 返回类型
module (nn.Module)
示例
>>> m = prune.identity(nn.Linear(2, 3), 'bias') >>> print(m.bias_mask) tensor([1., 1., 1.])