torch.nn.utils.parametrize.is_parametrized

torch.nn.utils.parametrize.is_parametrized(module, tensor_name=None) [source]

Возвращает True , если модуль имеет активную параметризацию.

Если аргумент tensor_name указан, возвращает True , если module[tensor_name] параметризован.

Параметры

module (nn.Module) – модуль для запроса
tensor_name (str, необязательно) – атрибут в модуле для запроса Значение по умолчанию: None

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