СлучайноеНеструктурированное

class torch.nn.utils.prune.RandomUnstructured(amount) [source]

Случайное удаление (в настоящее время не обрезанных) единиц в тензоре.

Параметры:

• name (str) – имя параметра в module по которому будет проводиться обрезка.
• amount (int или float) – количество параметров для обрезки. Если float, должно быть в диапазоне от 0.0 до 1.0 и представлять долю параметров для обрезки. Если int, оно представляет собой абсолютное число параметров для обрезки.

classmethod apply(module, name, amount) [source]

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

Параметры:

• module (nn.Module) – модуль, содержащий тензор для обрезки
• name (str) – имя параметра в module по которому будет проводиться обрезка.
• amount (int или float) – количество параметров для обрезки. Если float, должно быть в диапазоне от 0.0 до 1.0 и представлять долю параметров для обрезки. Если int, оно представляет собой абсолютное число параметров для обрезки.

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, необязательно) – маска с предыдущей итерации обрезки, если она есть. Для учета при определении части тензора, на которую должна действовать обрезка. Если None, используется маска из единиц.

Возвращает:

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

remove(module)

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

Примечание

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