Skip to content
Draft
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions docs/api/paddle/autograd/Overview_cn.rst
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@ paddle.autograd 目录下包含飞桨框架支持的自动微分相关的 API
:widths: 10, 30

" :ref:`backward <cn_api_paddle_autograd_backward>` ", "计算给定的 Tensors 的反向梯度"
" :ref:`enable_grad <cn_api_paddle_autograd_enable_grad>` ", "创建启用动态图梯度计算的上下文"
" :ref:`hessian <cn_api_paddle_autograd_hessian>` ", "计算因变量 ``ys`` 对 自变量 ``xs`` 的海森矩阵"
" :ref:`jacobian <cn_api_paddle_autograd_jacobian>` ", "计算因变量 ``ys`` 对 自变量 ``xs`` 的雅可比矩阵"
" :ref:`saved_tensors_hooks <cn_api_paddle_autograd_saved_tensors_hooks>` ", "用于动态图中为保存的 Tensor 注册一对 pack / unpack hook"
Expand Down
18 changes: 18 additions & 0 deletions docs/api/paddle/autograd/enable_grad_cn.rst
Original file line number Diff line number Diff line change
@@ -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
Original file line number Diff line number Diff line change
Expand Up @@ -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`` 命名空间下调用,具体如下:

### 转写示例

Expand All @@ -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

Expand Down