PropagateUnbackedSymInts

class torch.fx.experimental.symbolic_shapes.PropagateUnbackedSymInts(module, garbage_collect_values=True, graph=None) [исходный код]

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

Выполняет module посредством интерпретации и возвращает результат. Используется соглашение о вызове «boxed», при котором передаётся список аргументов, очищаемый интерпретатором. Это обеспечивает своевременное освобождение входных тензоров.

Примечание

Обратная совместимость этого API гарантируется.

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

Any

call_function(target, args, kwargs) [исходный код]

Выполняет узел call_function и возвращает результат.

Параметры:

• target (Target) – Цель вызова для этого узла. Подробные сведения о семантике см. в описании Node
• args (Tuple) – Кортеж позиционных аргументов этого вызова
• kwargs (Dict) – Словарь именованных аргументов этого вызова

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

Any

Возвращаемое значение

Any: значение, возвращённое вызовом функции

Примечание

Обратная совместимость этого API гарантируется.

call_method(target, args, kwargs) [исходный код]

Выполняет узел call_method и возвращает результат.

Параметры:

• target (Target) – Цель вызова для этого узла. Подробные сведения о семантике см. в описании Node
• args (Tuple) – Кортеж позиционных аргументов этого вызова
• kwargs (Dict) – Словарь именованных аргументов этого вызова

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

Any

Возвращаемое значение

Any: значение, возвращённое вызовом метода

Примечание

Обратная совместимость этого API гарантируется.

call_module(target, args, kwargs) [исходный код]

Выполняет узел call_module и возвращает результат.

Параметры:

• target (Target) – Цель вызова для этого узла. Подробные сведения о семантике см. в описании Node
• args (Tuple) – Кортеж позиционных аргументов этого вызова
• kwargs (Dict) – Словарь именованных аргументов этого вызова

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

Any

Возвращаемое значение

Any: значение, возвращённое вызовом модуля

Примечание

Обратная совместимость этого API гарантируется.

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

Извлекает конкретные значения args и kwargs узла n из текущего окружения выполнения.

Параметры:

n (Node) – Узел, для которого следует получить args и kwargs.

Возвращает:

args и kwargs с конкретными значениями для n.

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

Tuple[Tuple, Dict]

Примечание

Обратная совместимость этого API гарантируется.

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

Извлекает атрибут из иерархии Module объекта self.module.

Параметры:

target (str) – Полное имя извлекаемого атрибута

Возвращает:

Значение атрибута.

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

Any

Примечание

Обратная совместимость этого API гарантируется.

get_attr(target, args, kwargs) [исходный код]

Выполняет узел get_attr. Извлекает значение атрибута из иерархии Module объекта self.module.

Параметры:

• target (Target) – Цель вызова для этого узла. Подробные сведения о семантике см. в описании Node
• args (Tuple) – Кортеж позиционных аргументов этого вызова
• kwargs (Dict) – Словарь именованных аргументов этого вызова

Возвращает:

Значение извлечённого атрибута

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

Any

Примечание

Обратная совместимость этого API гарантируется.

map_nodes_to_values(args, n) [исходный код]

Рекурсивно обходит args и находит конкретное значение каждого Node в текущем окружении выполнения.

Параметры:

• args (Argument) – Структура данных, в которой нужно найти конкретные значения
• n (Node) – Узел, которому принадлежит args. Используется только для вывода сведений об ошибке.

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

tuple[Argument, …] | Sequence[Argument] | Mapping[str, Argument] | slice | range | Node | str | int | float | bool | complex | dtype | Tensor | device | memory_format | layout | OpOverload | SymInt | SymBool | SymFloat | None

Примечание

Обратная совместимость этого API гарантируется.

output(target, args, kwargs) [исходный код]

Выполняет узел output. Фактически он просто извлекает значение, на которое ссылается узел output, и возвращает его.

Параметры:

• target (Target) – Цель вызова для этого узла. Подробные сведения о семантике см. в описании Node
• args (Tuple) – Кортеж позиционных аргументов этого вызова
• kwargs (Dict) – Словарь именованных аргументов этого вызова

Возвращает:

Возвращаемое значение, на которое ссылается выходной узел

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

Any

Примечание

Обратная совместимость этого API гарантируется.

placeholder(target, args, kwargs) [исходный код]

Выполняет узел placeholder. Обратите внимание, что метод изменяет состояние: Interpreter поддерживает внутренний итератор по аргументам, переданным в run, и этот метод возвращает следующий элемент итератора.

Параметры:

• target (Target) – Цель вызова для этого узла. Подробные сведения о семантике см. в описании Node
• args (Tuple) – Кортеж позиционных аргументов этого вызова
• kwargs (Dict) – Словарь именованных аргументов этого вызова

Возвращает:

Извлечённое значение аргумента.

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

Any

Примечание

Обратная совместимость этого API гарантируется.

run(*args, initial_env=None, enable_io_processing=True) [исходный код]

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

Параметры:

• *args (Any) – Аргументы модуля, который нужно выполнить, в позиционном порядке
• initial_env (Optional[Dict[Node, Any]]) – Необязательное начальное окружение для выполнения. Это словарь, сопоставляющий Node с любым значением. Например, его можно использовать для предварительного заполнения результатов некоторых Nodes, чтобы выполнить в интерпретаторе только частичное вычисление.
• enable_io_processing (bool) – Если значение истинно, перед использованием сначала обрабатываются входные и выходные данные с помощью функций process_inputs и process_outputs графа.

Возвращает:

Значение, возвращённое при выполнении модуля

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

Any

Примечание

Обратная совместимость этого API гарантируется.

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

Выполняет узел FX, распространяя привязки неподкреплённых символов на новый фиктивный тензор

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

Tensor | tuple[Tensor, …]