square

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

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

\[out = x^2\]

参数

  • x (Tensor) - 任意维度的 Tensor,支持的数据类型:int32,int64,float32,float64,complex64,complex128。别名 input

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

关键字参数

  • out (Tensor,可选) - 输出 Tensor,若不为 None,计算结果将保存在该 Tensor 中,默认值为 None

返回

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

代码示例

>>> import paddle

>>> x = paddle.to_tensor([-0.4, -0.2, 0.1, 0.3])
>>> out = paddle.square(x)
>>> print(out)
Tensor(shape=[4], dtype=float32, place=Place(cpu), stop_gradient=True,
[0.16000001, 0.04000000, 0.01000000, 0.09000000])

使用本API的教程文档