From ca8e6add006b3fcfe7dd4663cefa76d1fd6afd3d Mon Sep 17 00:00:00 2001 From: zhaoyuchen2018 <45989343+zhaoyuchen2018@users.noreply.github.com> Date: Wed, 25 Sep 2019 17:35:01 +0800 Subject: [PATCH] Refine square api doc (#1273) test=develop Signed-off-by: zhaoyuchen --- doc/fluid/api_cn/layers_cn/square_cn.rst | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/doc/fluid/api_cn/layers_cn/square_cn.rst b/doc/fluid/api_cn/layers_cn/square_cn.rst index 0c6678bae..eef916215 100644 --- a/doc/fluid/api_cn/layers_cn/square_cn.rst +++ b/doc/fluid/api_cn/layers_cn/square_cn.rst @@ -5,24 +5,26 @@ square .. py:function:: paddle.fluid.layers.square(x,name=None) -取平方激活函数。 +该OP执行逐元素取平方运算。 .. math:: out = x^2 参数: - - **x** : 平方操作符的输入 - - **use_cudnn** (BOOLEAN) – （bool，默认为false）是否仅用于cudnn核，需要安装cudnn + - **x** (Variable) - 任意维度的Tensor，支持的数据类型： float32，float64。 + - **name** (str，可选) - 该参数供开发人员打印调试信息时使用，具体用法请参见 :ref:`api_guide_Name`，默认值为None。 -返回：平方后的结果 +返回：返回取平方后的Tensor，维度和数据类型同输入一致。 + +返回类型：Variable **代码示例**： .. code-block:: python import paddle.fluid as fluid - data = fluid.layers.data(name="input", shape=[32, 784]) - result = fluid.layers.square(data) + data = fluid.layers.data(name="input", shape=[32, 784], dtype='float32') + result = fluid.layers.square(data) #result.shape=[32, 784], type=float32 -- GitLab