From fb3ebab67cf6c375cd1ffc87c161a13b45c20ef9 Mon Sep 17 00:00:00 2001 From: neu-hsc <67090824@qq.com> Date: Fri, 19 Jun 2026 01:47:51 +0800 Subject: [PATCH] Document paddle.autograd.enable_grad --- docs/api/paddle/autograd/Overview_cn.rst | 1 + docs/api/paddle/autograd/enable_grad_cn.rst | 18 ++++++++++++++++++ .../torch.autograd.enable_grad.md | 8 ++++---- 3 files changed, 23 insertions(+), 4 deletions(-) create mode 100644 docs/api/paddle/autograd/enable_grad_cn.rst diff --git a/docs/api/paddle/autograd/Overview_cn.rst b/docs/api/paddle/autograd/Overview_cn.rst index 04568376fda..cd354ad7496 100644 --- a/docs/api/paddle/autograd/Overview_cn.rst +++ b/docs/api/paddle/autograd/Overview_cn.rst @@ -19,6 +19,7 @@ paddle.autograd 目录下包含飞桨框架支持的自动微分相关的 API :widths: 10, 30 " :ref:`backward ` ", "计算给定的 Tensors 的反向梯度" + " :ref:`enable_grad ` ", "创建启用动态图梯度计算的上下文" " :ref:`hessian ` ", "计算因变量 ``ys`` 对 自变量 ``xs`` 的海森矩阵" " :ref:`jacobian ` ", "计算因变量 ``ys`` 对 自变量 ``xs`` 的雅可比矩阵" " :ref:`saved_tensors_hooks ` ", "用于动态图中为保存的 Tensor 注册一对 pack / unpack hook" diff --git a/docs/api/paddle/autograd/enable_grad_cn.rst b/docs/api/paddle/autograd/enable_grad_cn.rst new file mode 100644 index 00000000000..e75430f7d72 --- /dev/null +++ b/docs/api/paddle/autograd/enable_grad_cn.rst @@ -0,0 +1,18 @@ +.. _cn_api_paddle_autograd_enable_grad: + +enable_grad +------------------------------- + +.. py:class:: paddle.autograd.enable_grad() + + +创建一个上下文来启用动态图梯度计算。在此模式下,每次计算的结果都将具有 stop_gradient=False。 + +也可以用作一个装饰器(需要创建实例对象作为装饰器)。 + +该 API 与 :ref:`cn_api_paddle_enable_grad` 功能一致。 + +代码示例 +:::::::::::: + +COPY-FROM: paddle.autograd.enable_grad diff --git a/docs/guides/model_convert/convert_from_pytorch/api_difference/invok_only_diff/torch.autograd.enable_grad.md b/docs/guides/model_convert/convert_from_pytorch/api_difference/invok_only_diff/torch.autograd.enable_grad.md index e2956628115..dadfd823b06 100644 --- a/docs/guides/model_convert/convert_from_pytorch/api_difference/invok_only_diff/torch.autograd.enable_grad.md +++ b/docs/guides/model_convert/convert_from_pytorch/api_difference/invok_only_diff/torch.autograd.enable_grad.md @@ -6,13 +6,13 @@ torch.autograd.enable_grad(*args, **kwargs) ``` -### [paddle.enable\_grad](https://www.paddlepaddle.org.cn/documentation/docs/zh/develop/api/paddle/enable_grad_cn.html#paddle.enable_grad) +### [paddle.autograd.enable\_grad](https://www.paddlepaddle.org.cn/documentation/docs/zh/develop/api/paddle/autograd/enable_grad_cn.html#paddle.autograd.enable_grad) ```python -paddle.enable_grad(*args, **kwargs) +paddle.autograd.enable_grad(*args, **kwargs) ``` -两者功能一致,但调用方式不一致,具体如下: +两者功能一致,Paddle 也支持在 ``paddle.autograd`` 命名空间下调用,具体如下: ### 转写示例 @@ -26,7 +26,7 @@ with torch.no_grad(): result = doubler(x) # Paddle 写法 -@paddle.enable_grad() +@paddle.autograd.enable_grad() def doubler(x): return x * 2