square¶

paddle. square ( x, name=None ) [source]

Square each elements of the inputs.

\[out = x^2\]

Parameters

• x (Tensor) – Input of Square operator, an N-D Tensor, with data type float32, float64, float16, complex64 or complex128.

• name (str, optional) – Name for the operation (optional, default is None). For more information, please refer to Name.

Returns

Tensor. Output of Square operator, a Tensor with shape same as input.

Examples

>>> 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])