FilesLast commitLast update
11 days ago
22 days ago
3 months ago
22 days ago
17 days ago
24 days ago
26 days ago
26 days ago
26 days ago
24 days ago
22 days ago
README

Equal

产品支持情况

产品 是否支持
Ascend 950PR&950DT系列产品 √
Atlas A3系列产品 √
Atlas A2系列产品 √
Atlas 200I/500 A2推理产品 ×
Atlas推理系列产品 √
Atlas训练系列产品 √

功能说明

  • 算子功能:逐元素比较两个输入张量是否相等,返回bool类型张量。

  • 计算表达式:

    out=(self==other)?True:Falseout = (self == other) ? True : False

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x1 输入 公式中的输入self。 FLOAT、BFLOAT16、FLOAT16、INT32、INT8、UINT8、DOUBLE、INT16、INT64、COMPLEX64、COMPLEX128、QUINT8、QINT8、QINT32、STRING、BOOL ND
x2 输入 公式中的输入other。 FLOAT、BFLOAT16、FLOAT16、INT32、INT8、UINT8、DOUBLE、INT16、INT64、COMPLEX64、COMPLEX128、QUINT8、QINT8、QINT32、STRING、BOOL ND
y 输出 公式中的out。 BOOL ND

约束说明

如果计算量过大可能会导致算子执行超时(aicore error类型报错,errorStr为:timeout or trap error),场景为最后2轴合轴小于16,前面的轴合轴超大。

调用说明

调用方式 调用样例 说明
aclnn API test_aclnn_eq_tensor.cpp 通过aclnnEqTensor接口方式调用Equal算子。
GE图模式 test_geir_equal.cpp 通过算子IR构图方式调用Equal算子。