快捷方式

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)[source][source]

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

返回类型

Callable[…, GraphModule]

文档

查阅 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获得问题解答

查看资源