PropagateUnbackedSymInts

class torch.fx.experimental.symbolic_shapes.PropagateUnbackedSymInts(module, garbage_collect_values=True, graph=None)

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)

從當前執行環境中獲取節點 n 的 args 和 kwargs 的具體值。

引數

n (Node) – 應獲取其 args 和 kwargs 的節點。這僅用於錯誤報告。

返回

包含節點 n 具體值的 args 和 kwargs。

返回型別

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[‘Argument’, …], Sequence[Argument], Mapping[str, Argument], 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) – 此呼叫的關鍵字引數字典

返回

output 節點引用的返回值

返回型別

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 – 要執行的 Module 的引數，按位置順序排列

• initial_env (Optional[Dict[Node, Any]]) – 可選的執行起始環境。這是一個將 Node 對映到任意值的字典。例如，這可以用於為某些 Node 預填充結果，以便在直譯器內僅進行部分評估。

• enable_io_processing (bool) – 如果為 true，我們首先使用 graph 的 process_inputs 和 process_outputs 函式處理輸入和輸出，然後再使用它們。

返回

執行 Module 返回的值

返回型別

Any

注意

此 API 的向後相容性得到保證。

run_node(n)

執行一個 FX 節點，將未支援的 Symbol 繫結傳播到新的 fake tensor

返回型別

Union[Tensor, tuple[torch.Tensor, …]]