sqrt_cn.rst 1.1 KB
Newer Older
H
Hao Wang 已提交
1 2 3 4 5 6 7
.. _cn_api_fluid_layers_sqrt:

sqrt
-------------------------------

.. py:function:: paddle.fluid.layers.sqrt(x, name=None)

8
计算输入的算数平方根。
H
Hao Wang 已提交
9

10
.. math:: out=\sqrt x=x^{1/2}
H
Hao Wang 已提交
11

12
<font color="#FF0000">**注意:请确保输入中的数值是非负数。**</font>
H
Hao Wang 已提交
13 14 15

参数:

16
    - **x** (Variable) - 支持任意维度的Tensor。数据类型为float32,float64或float16。
P
Pei Yang 已提交
17
    - **name** (str,可选) – 具体用法请参见 :ref:`api_guide_Name` ,一般无需设置,默认值为None。
H
Hao Wang 已提交
18

19
返回:返回类型为Variable(Tensor|LoDTensor), 数据类型同输入一致。
H
Hao Wang 已提交
20 21 22 23 24

**代码示例**:

.. code-block:: python

25
        import numpy as np
H
Hao Wang 已提交
26
        import paddle.fluid as fluid
27 28 29 30 31 32 33 34 35 36 37 38

        inputs = fluid.layers.data(name="x", shape = [3], dtype='float32')
        output = fluid.layers.sqrt(inputs)

        exe = fluid.Executor(fluid.CPUPlace())
        exe.run(fluid.default_startup_program())

        img = np.array([0, 9, 36]).astype(np.float32)

        res = exe.run(fluid.default_main_program(), feed={'x':img}, fetch_list=[output])
        print(res)
        # [array([0., 3., 6.], dtype=float32)] 
H
Hao Wang 已提交
39 40 41 42 43 44 45 46 47 48 49 50 51