CodeGen

class torch.fx.graph.CodeGen [исходный код]

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

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

additional_globals() [исходный код]

Если ваш генератор кода использует дополнительные глобальные значения, добавьте сюда кортежи (идентификатор, ссылка на значение). Например, верните [‘List’, typing.List], если вам нужен List в глобальном контексте.

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

list[tuple[str, Any]]

gen_fn_def(free_vars, maybe_return_annotation, *, expanded_def=False) [исходный код]

На основе свободных переменных и аннотации возвращаемого значения генерирует начало функции FX. По умолчанию gen_fn_def([‘a’, ‘b’], ‘’) == ‘def {self._func_name}(a, b):’

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

str

generate_output(output_args, *, descs=None, repr_fn=None) [исходный код]

На основе аргументов выходных данных генерирует оператор return функции FX. Примечание: возвращаемый оператор не должен иметь отступа.

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

str

process_inputs(*args) [исходный код]

Преобразует входные данные так, чтобы граф мог принимать их в качестве аргументов, поскольку генератор кода, отличный от стандартного, может привести к тому, что входные данные функции будут отличаться от входных данных графа.

Если граф можно было запускать напрямую, этот инвариант должен выполняться f.graph.process_outputs(f.graph(*f.graph.process_inputs(*inputs))) == f(*inputs)

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

Any

process_outputs(outputs) [исходный код]

Преобразует выходные данные графа так, чтобы они совпадали с результатом генерации кода.

Подробнее см. process_inputs.

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

Any