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)