torch.fx.operator_schemas.get_signature_for_torch_op

torch.fx.operator_schemas.get_signature_for_torch_op(op: Callable[[...], Any], return_schemas: Literal[True]) → tuple[list[Signature] | None, list[FunctionSchema] | None] [source]

torch.fx.operator_schemas.get_signature_for_torch_op(op:Callable[[...],Any], return_schemas:Literal[False]=False) → list[Signature]|None

Для оператора в пространстве имён torch возвращает список объектов inspect.Signature, соответствующих перегрузкам этого оператора. Может вернуть None, если получить сигнатуру не удалось.

Параметры:

op (Callable) – Оператор в пространстве имён torch, для которого нужно найти сигнатуру

Возвращает:

Список сигнатур перегрузок этого

оператора или None, если получить сигнатуры оператора не удалось. Если return_schemas=True, возвращает кортеж, содержащий необязательные сигнатуры Python и необязательную сигнатуру функции TorchScript

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

Optional[List[inspect.Signature]]

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

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