sqrt

  • paddle.fluid.layers.sqrt(x, name=None)

计算输入的算数平方根。

sqrt - 图1

注解

请确保输入中的数值是非负数。

参数:

  • x (Variable) - 支持任意维度的Tensor。数据类型为float32,float64或float16。
  • name (str,可选) – 具体用法请参见 Name ,一般无需设置,默认值为None。

返回:返回类型为Variable(Tensor|LoDTensor), 数据类型同输入一致。

代码示例

  1. import numpy as np
  2. import paddle.fluid as fluid
  3.  
  4. inputs = fluid.layers.data(name="x", shape = [3], dtype='float32')
  5. output = fluid.layers.sqrt(inputs)
  6.  
  7. exe = fluid.Executor(fluid.CPUPlace())
  8. exe.run(fluid.default_startup_program())
  9.  
  10. img = np.array([0, 9, 36]).astype(np.float32)
  11.  
  12. res = exe.run(fluid.default_main_program(), feed={'x':img}, fetch_list=[output])
  13. print(res)
  14. # [array([0., 3., 6.], dtype=float32)]