评价此页

torch.fx.experimental.proxy_tensor.get_proxy_mode#

torch.fx.experimental.proxy_tensor.get_proxy_mode()[source]#

当前活动的代理跟踪模式,如果当前没有跟踪则为 None。这包括预分派代理跟踪。

返回类型

Optional[ProxyTorchDispatchMode]