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) – 此調用的位置引數 tuple

• kwargs (Dict) – 此調用的關鍵字引數 dict

回傳類型

Any

回傳

Any: 函數調用所回傳的值

注意

此 API 保證向後相容性。

call_method(target, args, kwargs)

執行一個 call_method 節點並回傳結果。

參數

• target (Target) – 此節點的呼叫目標。 請參閱 Node 以取得語意詳細資訊。

• args (Tuple) – 此調用的位置引數 tuple

• kwargs (Dict) – 此調用的關鍵字引數 dict

回傳類型

Any

回傳

Any: 方法調用所回傳的值

注意

此 API 保證向後相容性。

call_module(target, args, kwargs)

執行一個 call_module 節點並回傳結果。

參數

• target (Target) – 此節點的呼叫目標。 請參閱 Node 以取得語意詳細資訊。

• args (Tuple) – 此調用的位置引數 tuple

• kwargs (Dict) – 此調用的關鍵字引數 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) – 此調用的位置引數 tuple

• kwargs (Dict) – 此調用的關鍵字引數 dict

回傳

已檢索的屬性的值

回傳類型

Any

注意

此 API 保證向後相容性。

map_nodes_to_values(args, n)

以遞迴方式遍歷 args 並在目前的執行環境中查詢每個 Node 的具體值。

參數

• args (Argument) – 要在其中查詢具體值的資料結構

• n (Node) – args 所屬的節點。 這僅用於錯誤報告。

回傳類型

Optional[Union[Tuple[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]], …], Sequence[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]]], Mapping[str, 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]]], 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) – 此調用的位置引數 tuple

• kwargs (Dict) – 此調用的關鍵字引數 dict

回傳

輸出節點所引用的傳回值

回傳類型

Any

注意

此 API 保證向後相容性。

placeholder(target, args, kwargs)

執行一個 placeholder 節點。 請注意，這是具狀態的：Interpreter 維護一個內部迭代器，遍歷傳遞給 run 的參數，而此方法會針對該迭代器傳回 next()。

參數

• target (Target) – 此節點的呼叫目標。 請參閱 Node 以取得語意詳細資訊。

• args (Tuple) – 此調用的位置引數 tuple

• kwargs (Dict) – 此調用的關鍵字引數 dict

回傳

檢索到的參數值。

回傳類型

Any

注意

此 API 保證向後相容性。

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

透過解譯執行 module 並傳回結果。

參數

• *args – 要執行的 Module 的引數，依位置順序排列

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

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

回傳

從執行 Module 傳回的值

回傳類型

Any

注意

此 API 保證向後相容性。

run_node(n)

執行一個 FX 節點，將未備份的 Symbol 綁定傳播到新的假張量

回傳類型

Union[Tensor, Tuple[Tensor, …]]