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

Y
Youwei Song 已提交
12 13
.. note::
    请确保输入中的数值是非负数。
H
Hao Wang 已提交
14 15 16

参数:

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

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

**代码示例**:

.. code-block:: python

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

        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 已提交
40 41 42 43 44 45 46 47 48 49 50 51 52