GraphAppendingTracer

class torch.fx.proxy.GraphAppendingTracer(graph) [исходный код]

Примечание

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

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

Метод, который преобразует объекты, встречающиеся в качестве аргументов при символьном вычислении, в типы Argument, которые можно сохранить в IR.

Метод можно переопределить, чтобы поддержать дополнительные типы, специфичные для трассировки.

Примечание

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

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

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

create_node(kind, target, args, kwargs, name=None, type_expr=None) [исходный код]

Вставляет узел графа с заданными target, args, kwargs и name.

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

Примечание

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

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

Node

create_proxy(kind, target, args, kwargs, name=None, type_expr=None, proxy_factory_fn=None) [исходный код]

Создаёт Node из заданных аргументов, а затем возвращает Node, обёрнутый в объект Proxy.

Если kind = ‘placeholder’, создаётся Node, представляющий параметр функции. Если необходимо закодировать параметр по умолчанию, используется кортеж args. В противном случае args для узлов placeholder остаётся пустым.

Примечание

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

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

Proxy

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

Вызывается при переборе объекта proxy, например,

при использовании в потоке управления. Обычно неизвестно, что делать, поскольку значение proxy неизвестно, однако пользовательский трассировщик может добавить дополнительные сведения к узлу графа с помощью create_node и вернуть итератор.

Примечание

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

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

Iterator

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

Вызывается при вызове метода keys() у объекта proxy.

Так происходит при вызове ** для объекта proxy. Если в вашем пользовательском трассировщике должен поддерживаться оператор **, этот метод должен возвращать итератор.

Примечание

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

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

Proxy

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

Примечание

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

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

Proxy

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

Вызывается при преобразовании объекта proxy в логическое значение, например,

при использовании в потоке управления. Обычно неизвестно, что делать, поскольку значение proxy неизвестно, однако пользовательский трассировщик может добавить дополнительные сведения к узлу графа с помощью create_node и вернуть значение.

Примечание

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

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

bool