square

paddle.fluid.layers.square(x, name=None)

The OP square each elements of the inputs.

\(out = x^2\)

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

  • name (str, optional) – The default value is None. Normally there is no need for user to set this property. For more information, please refer to Name .

Returns

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

Return type

Variable

Examples

import paddle.fluid as fluid
import numpy as np

inputs = fluid.data(name="x", shape = [None, 4], dtype='float32')
output = fluid.layers.square(inputs)

exe = fluid.Executor(fluid.CPUPlace())
exe.run(fluid.default_startup_program())

#input.shape=1X4, batch_size=1
img = np.array([[1.0, 2.0, 3.0, 4.0]]).astype(np.float32)
res = exe.run(fluid.default_main_program(), feed={'x':img}, fetch_list=[output])
print(res)