[ 仅 API 调用方式不一致 ]torch.autograd.enable_grad

torch.autograd.enable_grad(*args, **kwargs)

paddle.enable_grad(*args, **kwargs)

# PyTorch 写法
@torch.autograd.enable_grad()
def doubler(x):
    return x * 2

with torch.no_grad():
    result = doubler(x)

# Paddle 写法
@paddle.enable_grad()
def doubler(x):
    return x * 2

with paddle.no_grad():
    result = doubler(x)