快捷方式

PropagateUnbackedSymInts

class torch.fx.experimental.symbolic_shapes.PropagateUnbackedSymInts(module, garbage_collect_values=True, graph=None)[source]
boxed_run(args_list)

通过解释运行 module 并返回结果。这使用“boxed”调用约定,您传递一个参数列表,该列表将在解释器中清除。这确保输入张量会立即释放。

注意

此 API 的向后兼容性得到保证。

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)

从当前执行环境中获取节点 nargskwargs 的具体值。

参数

n (Node) – 应该为其获取 argskwargs 的节点。

返回值

argskwargs,其中包含 n 的具体值。

返回类型

Tuple[Tuple, Dict]

注意

此 API 的向后兼容性得到保证。

fetch_attr(target)

self.moduleModule 层次结构中获取属性。

参数

target (str) – 要获取的属性的完全限定名称

返回值

属性的值。

返回类型

Any

注意

此 API 的向后兼容性得到保证。

get_attr(target, args, kwargs)

执行 get_attr 节点。 将从 self.moduleModule 层级结构中检索属性值。

参数
  • target (Target) – 此节点的调用目标。有关语义的详细信息,请参见 Node

  • args (Tuple) – 此调用的位置参数元组

  • kwargs (Dict) – 此调用的关键字参数字典

返回值

检索到的属性值

返回类型

Any

注意

此 API 的向后兼容性得到保证。

map_nodes_to_values(args, n)

递归地遍历 args 并查找当前执行环境中每个 Node 的具体值。

参数
  • args (Argument) – 用于查找具体值的数据结构

  • n (Node) – args 所属的节点。 这仅用于错误报告。

返回类型

Optional[Union[Tuple[Any, …], List[Any], Dict[str, Any], slice, range, Node, str, int, float, bool, complex, dtype, Tensor, device, memory_format, layout, OpOverload, SymInt, SymBool, SymFloat]]

注意

此 API 的向后兼容性得到保证。

output(target, args, kwargs)

执行 output 节点。 这实际上只是检索由 output 节点引用的值并将其返回。

参数
  • target (Target) – 此节点的调用目标。有关语义的详细信息,请参见 Node

  • args (Tuple) – 此调用的位置参数元组

  • kwargs (Dict) – 此调用的关键字参数字典

返回值

输出节点引用的返回值

返回类型

Any

注意

此 API 的向后兼容性得到保证。

placeholder(target, args, kwargs)

执行 placeholder 节点。 请注意,这是有状态的:Interpreter 保持对传递给 run 的参数的内部迭代器,此方法返回该迭代器上的 next()。

参数
  • target (Target) – 此节点的调用目标。有关语义的详细信息,请参见 Node

  • args (Tuple) – 此调用的位置参数元组

  • kwargs (Dict) – 此调用的关键字参数字典

返回值

检索到的参数值。

返回类型

Any

注意

此 API 的向后兼容性得到保证。

run(*args, initial_env=None, enable_io_processing=True)

通过解释运行 module 并返回结果。

参数
  • *args – 要运行的模块的参数,按位置顺序

  • initial_env (Optional[Dict[Node, Any]]) – 可选的执行起始环境。 这是一个字典,将 Node 映射到任何值。 例如,这可用于为某些 Nodes 预先填充结果,以便在解释器中仅进行部分评估。

  • enable_io_processing (bool) – 如果为 True,我们将在使用输入和输出之前,先使用图的 process_inputs 和 process_outputs 函数对其进行处理。

返回值

执行模块后返回的值

返回类型

Any

注意

此 API 的向后兼容性得到保证。

run_node(n)[source]

运行 FX 节点,将未支持的符号绑定传播到新的伪张量

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的答案

查看资源