torch.nn.utils.parametrize.type_before_parametrizations

torch.nn.utils.parametrize.type_before_parametrizations(module) [исходный код]

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

Параметры:

module (nn.Module) – модуль, тип которого нужно получить

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

type