equal_cn.rst 1.5 KB
Newer Older
H
Hao Wang 已提交
1 2 3 4 5
.. _cn_api_fluid_layers_equal:

equal
-------------------------------

6
.. py:function:: paddle.fluid.layers.equal(x, y, cond=None, name=None)
S
swtkiwi 已提交
7 8


9
该OP返回 :math:`x==y` 逐元素比较x和y是否相等,x和y的维度应该相同。
H
Hao Wang 已提交
10 11

参数:
12 13
    - **x** (Variable) - 输入Tensor,支持的数据类型包括 float32, float64,int32, int64。
    - **y** (Variable) - 输入Tensor,支持的数据类型包括 float32, float64, int32, int64。
14 15
    - **cond** (Variable,可选) – 如果为None,则创建一个Tensor来作为进行比较的输出结果,该Tensor的shape和数据类型和输入x一致;如果不为None,则将Tensor作为该OP的输出,数据类型和数据shape需要和输入x一致。默认值为None。 
    - **name** (str,可选)- 具体用法请参见 :ref:`api_guide_Name` ,一般无需设置,默认值为None。
H
Hao Wang 已提交
16

17
返回:输出结果的Tensor,输出Tensor的shape和输入一致,Tensor数据类型为bool。
H
Hao Wang 已提交
18 19 20 21 22 23 24 25

返回类型:变量(Variable)

**代码示例**:

.. code-block:: python

    import paddle.fluid as fluid
26
    import numpy as np
W
wawltor 已提交
27
    
W
wangchaochaohu 已提交
28
    out_cond =fluid.data(name="input1", shape=[2], dtype='bool')
29 30
    label = fluid.layers.assign(np.array([3, 3], dtype="int32"))
    limit = fluid.layers.assign(np.array([3, 2], dtype="int32"))
W
wawltor 已提交
31 32 33 34 35 36 37
    label_cond = fluid.layers.assign(np.array([1, 2], dtype="int32"))
    
    out1 = fluid.layers.equal(x=label,y=limit) #out1=[True, False]
    out2 = fluid.layers.equal(x=label_cond,y=limit, cond=out_cond) #out2=[False, True] out_cond=[False, True]
    


H
Hao Wang 已提交
38 39