square¶

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

The OP 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.

• with_quant_attr (BOOLEAN) – Whether the operator has attributes used by quantization.

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

Returns

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

Return type

out (Tensor)

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]