TensorEqual
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR/Ascend 950DT | √ |
| Atlas A3 训练系列产品/Atlas A3 推理系列产品 | √ |
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 | √ |
| Atlas 200I/500 A2 推理产品 | × |
| Atlas 推理系列产品 | √ |
| Atlas 训练系列产品 | √ |
功能说明
-
接口功能:计算两个Tensor是否有相同的大小和元素,返回一个Bool类型。
-
计算表达式:
out=(self==other)?True:Falseout = (self == other) ? True : False
参数说明
| 参数名 | 输入/输出 | 描述 | 使用说明 | 数据类型 | 数据格式 | 维度(shape) | 非连续Tensor |
|---|---|---|---|---|---|---|---|
| self | 输入 | 表示第一个输入。 | self与other的数据类型满足数据类型推导规则(参见互推导关系)。 | FLOAT16、FLOAT、INT32、INT8、UINT8、BOOL、DOUBLE、INT64、INT16、UINT16、UINT32、UINT64、BFLOAT16 | ND | - | √ |
| other | 输入 | 表示第二个输入。 | other与self的数据类型满足数据类型推导规则(参见互推导关系)。 | FLOAT16、FLOAT、INT32、INT8、UINT8、BOOL、DOUBLE、INT64、INT16、UINT16、UINT32、UINT64、BFLOAT16 | ND | - | √ |
| out | 输出 | 表示输出。输出一个数据类型为BOOL,一维包含一个元素的Tensor。 | - | - | - | - | - |
- Atlas 推理系列产品、Atlas 训练系列产品:数据类型支持FLOAT16、FLOAT、DOUBLE。
约束说明
无
调用说明
| 调用方式 | 样例代码 | 说明 |
|---|---|---|
| aclnn调用 | test_aclnn_equal.cpp | 通过aclnnEqual接口方式调用TensorEqual算子。 |