square

paddle. square ( x, name=None ) [源代码]

对输入参数``x``进行逐元素取平方运算。

\[out = x^2\]

参数

  • x (Tensor) - 任意维度的 Tensor,支持的数据类型:float32,float64。

  • name (str,可选) - 具体用法请参见 Name,一般无需设置,默认值为 None。

返回

取平方后的 Tensor,维度和数据类型同输入一致。

代码示例

import paddle

x = paddle.to_tensor([-0.4, -0.2, 0.1, 0.3])
out = paddle.square(x)
print(out)
# [0.16 0.04 0.01 0.09]