torch.nn.utils.prune.identity

torch.nn.utils.prune.identity(module, name) [исходный код]

Применяет параметризацию прореживания, не прореживая ни одной единицы.

Применяет параметризацию прореживания к тензору, соответствующему параметру с именем name в module, не прореживая фактически ни одной единицы. Изменяет модуль на месте (и также возвращает изменённый модуль), выполняя следующие действия:

1. добавляет именованный буфер с именем name+'_mask', соответствующий бинарной маске, применённой методом прореживания к параметру name.
2. заменяет параметр name его прореженной версией, сохраняя исходный (непрореженный) параметр в новом параметре с именем name+'_orig'.

Примечание

Маска представляет собой тензор из единиц.

Параметры:

• 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.])