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, stack_trace=False)[source]#

给定一个函数 f，返回一个新函数，该函数在用 f 的有效参数执行时，会返回一个表示执行过程中执行的操作集的 FX GraphModule。

如果 stack_trace 为 True，则 stack_trace 将保留在 node.meta["stack_trace"] 中