square

paddle.square ( x, name=None )

该OP执行逐元素取平方运算。

参数:

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

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

返回：返回取平方后的Tensor，维度和数据类型同输入一致。

返回类型：Tensor

代码示例：

import paddle
x = paddle.rand(shape=[32, 784])
y = paddle.square(x)
print(y.shape)