square
paddle.fluid.layers.
square
(x, name=None)
该OP执行逐元素取平方运算。
- 参数:
- x (Variable) - 任意维度的Tensor,支持的数据类型: float32,float64。
- name (str,可选) - 具体用法请参见 Name ,一般无需设置,默认值为None。
返回:返回取平方后的Tensor,维度和数据类型同输入一致。
返回类型:Variable
代码示例:
- import paddle.fluid as fluid
- data = fluid.layers.data(name="input", shape=[32, 784], dtype='float32')
- result = fluid.layers.square(data) #result.shape=[32, 784], type=float32