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)[原始碼][原始碼]¶

給定一個函數 f，回傳一個新的函數，當使用 f 的有效參數執行時，會回傳一個 FX GraphModule，代表在執行過程中執行的運算集合。