FilesLast commitLast update
1 month ago
1 month ago
1 month ago
1 month ago
1 month ago
5 months ago
1 month ago
README

Equal

产品支持情况

产品 是否支持
Atlas A2系列产品 √
Atlas A3系列产品 √

功能说明

  • 算子功能:对两个输入张量进行逐元素相等比较,并根据广播后的形状输出BOOL张量。
  • 计算公式:

outi=(selfi==otheri)out_i = (self_i == other_i)

  • 浮点比较采用数值语义:+0.0与-0.0相等,NaN与任意值(包括NaN)均不相等,同号无穷大相等。

参数说明

参数名 输入/输出 描述 使用说明 数据类型 数据格式 维度(shape) 非连续Tensor
self 输入 待比较的第一个输入张量。 数据类型与other一致;shape与other满足broadcast关系。 FLOAT16、FLOAT、BFLOAT16、INT8、UINT8、INT16、INT32、UINT32、INT64、BOOL ND 0-8 √
other 输入 待比较的第二个输入张量。 数据类型与self一致;shape与self满足broadcast关系。 FLOAT16、FLOAT、BFLOAT16、INT8、UINT8、INT16、INT32、UINT32、INT64、BOOL ND 0-8 √
out 输出 逐元素比较结果。 shape为self与other广播后的公共shape。 BOOL ND 0-8 √
workspaceSize 输出 返回需要在Device侧申请的workspace大小。 - - - - -
executor 输出 返回op执行器,包含算子计算流程。 - - - - -

约束说明

  • self与other的数据类型必须一致。
  • self与other的shape必须满足broadcast关系。
  • out的数据类型必须为BOOL,shape必须与输入广播后的公共shape一致。

调用说明

调用方式 调用样例 说明
aclnn调用 test_aclnn_eq_tensor.cpp 通过aclnnEqTensor/aclnnInplaceEqTensor接口调用Equal算子。
aclnn调用 test_aclnn_eq_scalar.cpp 通过aclnnEqScalar/aclnnInplaceEqScalar接口调用Equal算子。

贡献说明

贡献者 贡献方 贡献算子 贡献时间 贡献内容
gcw_rESKWmgp 个人开发者 Equal 2026/08/27 完成A2/A3 Ascend C实现、数据类型适配、广播支持及性能优化。