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)