no_grad_cn.rst 814 字节
Newer Older
Z
zq19 已提交
1 2 3 4 5
.. _cn_api_fluid_dygraph_no_grad:

no_grad
-------------------------------

6 7
**注意:该API仅支持【动态图】模式**

Z
zq19 已提交
8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25
.. py:method:: paddle.fluid.dygraph.no_grad(func)

在动态图模式中,此装饰器将会避免 ``func`` 被装饰时创建反向传播网络。

参数:
    - **func** (str) – 不需要梯度的函数。

**代码示例**

..  code-block:: python


    import numpy as np
    import paddle.fluid as fluid

    @fluid.dygraph.no_grad
    def test_layer():
        with fluid.dygraph.guard():
26
            inp = np.ones([3, 1024], dtype='float32')
Z
zq19 已提交
27
            t = fluid.dygraph.base.to_variable(inp)
28 29 30 31
            linear1 = fluid.Linear(1024, 4, bias_attr=False)
            linear2 = fluid.Linear(4, 4)
            ret = linear1(t)
            dy_ret = linear2(ret)
Z
zq19 已提交
32

33
    test_layer()