torch.is_grad_enabled

torch.is_grad_enabled()

如果当前启用 grad 模式，则返回 True。