torch.fx.operator_schemas.normalize_module

torch.fx.operator_schemas.normalize_module(root, target, args, kwargs=None, normalize_to_only_use_kwargs=False) [исходный код]

Возвращает нормализованные аргументы модулей PyTorch. Это означает, что args/kwargs будут сопоставлены с сигнатурой функциональной операции, а если normalize_to_only_use_kwargs имеет значение True, будут возвращены только kwargs в порядке позиционных аргументов. Также заполняются значения по умолчанию. Позиционные параметры с ограничением только по позиции и параметры varargs не поддерживаются (*args, **kwargs).

Параметры:

• root (nn.Module) – корневой модуль, в котором выполняется поиск модулей
• target (Callable) – функция, аргументы которой нормализуются
• args (Tuple[Any]) – кортеж аргументов функции
• kwargs (Optional[Dict[str, Any]]) – словарь kwargs функции
• normalize_to_only_use_kwargs (bool) – следует ли нормализовать аргументы так, чтобы использовать только kwargs.

Возвращает:

Возвращает normalized_args_and_kwargs или None в случае неудачи.

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

ArgsKwargsPair | None

Предупреждение

Этот API является экспериментальным и НЕ сохраняет обратную совместимость.