torch.fx.experimental.proxy_tensor.make_fx#

torch.fx.experimental.proxy_tensor.make_fx(f, decomposition_table=None, tracing_mode='real', _allow_non_fake_inputs=False, *, pre_dispatch=False, record_module_stack=False, _allow_fake_constant=False, _error_on_data_dependent_ops=True, record_stack_traces=False)[原始碼]#

給定一個函式 f，返回一個新函式，當使用 f 的有效引數執行該函式時，它會返回一個 FX GraphModule，表示在執行過程中執行的操作集。

如果 record_stack_traces 為 True，則會在 node.meta[“stack_trace”] 上保留堆疊跟蹤。