torch.fx.experimental.proxy_tensor.get_proxy_mode

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

Current the currently active proxy tracing mode, or None if we are not currently tracing. This includes pre-dispatch proxy tracing.

Return type

Optional[ProxyTorchDispatchMode]