Идентичность

class torch.nn.utils.prune.Identity [source]

Метод обрезки, который не обрезает никакие элементы, но генерирует параметризацию обрезки с маской из единиц.

classmethod apply(module, name) [source]

Добавляет предварительный хук в функцию forward, который позволяет производить обрезку на лету и перепараметризовать тензор в терминах исходного тензора и маски обрезки.

Параметры

• module (nn.Module) – модуль, содержащий тензор для обрезки
• name (str) – имя параметра внутри module, на котором будет производиться обрезка.

apply_mask(module)

Просто выполняет умножение параметра, который обрезается, на сгенерированную маску. Извлекает маску и исходный тензор из модуля и возвращает обрезанную версию тензора.

Параметры

module (nn.Module) – модуль, содержащий тензор для обрезки

Возвращает

обрезанная версия входного тензора

Тип возвращаемого значения

pruned_tensor (torch.Tensor)

prune(t, default_mask=None, importance_scores=None)

Вычисляет и возвращает обрезанную версию входного тензора t в соответствии с правилом обрезки, указанным в compute_mask().

Параметры

• t (torch.Tensor) – тензор для обрезки (того же размера, что и default_mask).
• importance_scores (torch.Tensor) – тензор оценок важности (того же размера, что и t) используемый для вычисления маски для обрезки t. Значения в этом тензоре указывают на важность соответствующих элементов в t который обрезается. Если не указано или равно None, используется тензор t.
• default_mask (torch.Tensor, optional) – маска из предыдущей итерации обрезки, если таковая имеется. Учитывается при определении части тензора, на которую должна повлиять обрезка. Если None, по умолчанию используется маска из единиц.

Возвращает

обрезанная версия тензора t.

remove(module)

Удаляет перепараметризацию обрезки из модуля. Обрезанный параметр с именем name остается постоянно обрезанным, а параметр с именем name+'_orig' удаляется из списка параметров. Аналогично, буфер с именем name+'_mask' удаляется из буферов.

Примечание

Сама обрезка НЕ отменяется или не восстанавливается!