InferenceMode ============= Context manager that enables or disables inference mode (gradient isolation). **Abstract Signature:** ``InferenceMode(mode: bool = True)`` .. raw:: html
torch.inference_mode
contextlib.nullcontext