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 or float16.

• 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)
# [0.16 0.04 0.01 0.09]