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: 函数调用返回的值
注意
此 API 的向后兼容性得到保证。
- call_method(target, args, kwargs)¶
执行
call_method
节点并返回结果。- 参数
target (Target) – 此节点的调用目标。有关语义的详细信息,请参见 Node
args (Tuple) – 此调用的位置参数元组
kwargs (Dict) – 此调用的关键字参数字典
- 返回类型
- 返回值
Any: 方法调用返回的值
注意
此 API 的向后兼容性得到保证。
- call_module(target, args, kwargs)¶
执行
call_module
节点并返回结果。- 参数
target (Target) – 此节点的调用目标。有关语义的详细信息,请参见 Node
args (Tuple) – 此调用的位置参数元组
kwargs (Dict) – 此调用的关键字参数字典
- 返回类型
- 返回值
Any: 模块调用返回的值
注意
此 API 的向后兼容性得到保证。
- fetch_args_kwargs_from_env(n)¶
从当前执行环境中获取节点
n
的args
和kwargs
的具体值。- 参数
n (Node) – 应该为其获取
args
和kwargs
的节点。- 返回值
args
和kwargs
,其中包含n
的具体值。- 返回类型
Tuple[Tuple, Dict]
注意
此 API 的向后兼容性得到保证。
- fetch_attr(target)¶
从
self.module
的Module
层次结构中获取属性。- 参数
target (str) – 要获取的属性的完全限定名称
- 返回值
属性的值。
- 返回类型
Any
注意
此 API 的向后兼容性得到保证。
- get_attr(target, args, kwargs)¶
执行
get_attr
节点。 将从self.module
的Module
层级结构中检索属性值。- 参数
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 并返回结果。
- 参数
- 返回值
执行模块后返回的值
- 返回类型
Any
注意
此 API 的向后兼容性得到保证。