torch.fx.operator_schemas.normalize_function

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

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

Для устранения неоднозначности перегрузок могут потребоваться arg_types и kwarg_types.

Параметры:

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

Возвращает:

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

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

ArgsKwargsPair | None

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

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