Python PyTorch identity用法及代碼示例

本文簡要介紹python語言中 torch.nn.utils.prune.identity 的用法。

用法:

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

參數：

• module(torch.nn.Module) -包含要修剪的張量的模塊。

• name(str) -module 中的參數名稱，將對其進行修剪。

返回：

輸入模塊的修改(即修剪)版本

返回類型：

模塊(nn.Module)

對與module 中名為name 的參數對應的張量應用修剪重新參數化，而不實際修剪任何單元。通過以下方式修改模塊(並返回修改後的模塊)：

1. 添加一個名為 name+'_mask' 的命名緩衝區，該緩衝區對應於通過修剪方法應用於參數 name 的二進製掩碼。

2. 將參數 name 替換為其修剪版本，而原始(未修剪)參數存儲在名為 name+'_orig' 的新參數中。

注意

掩碼是一個張量。

例子

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