torch.nn.utils.prune.random_unstructured¶
- torch.nn.utils.prune.random_unstructured(module, name, amount)[源代码]¶
通过删除随机(当前未修剪的)单元来修剪张量。
通过删除在随机位置选择的指定
amount
个(当前未修剪的)单元,修剪对应于module
中名为name
的参数的张量。通过以下方式对模块进行就地修改(并返回修改后的模块):添加一个名为
name+'_mask'
的命名缓冲区,对应于修剪方法应用于参数name
的二进制掩码。用其修剪后的版本替换参数
name
,而原始(未修剪的)参数存储在一个名为name+'_orig'
的新参数中。
- 参数
- 返回值
输入模块的修改版(即修剪后的版本)
- 返回类型
module (nn.Module)
示例
>>> m = prune.random_unstructured(nn.Linear(2, 3), 'weight', amount=1) >>> torch.sum(m.weight_mask == 0) tensor(1)