Atan2

产品支持情况

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

功能说明

• 算子功能：对输入张量self和other进行逐元素的反正切运算，self表示y坐标，other表示x坐标，计算结果为点(other, self)与x轴正方向的夹角（单位为弧度），取值范围为[-π, π]，支持广播。

• 计算公式：

  $$ou{t}_i=atan2(sel{f}_i,othe{r}_i)$$

  其中，$sel{f}_i$表示输入张量self的第$i$个元素，$othe{r}_i$表示输入张量other的第$i$个元素。计算结果根据(other, self)所在象限确定：

  $$ou{t}_i=\{\begin{array}{ll}atan(sel{f}_i/othe{r}_i)+\pi & \text{if }othe{r}_i<0,sel{f}_i\ge 0\\ atan(sel{f}_i/othe{r}_i)-\pi & \text{if }othe{r}_i<0,sel{f}_i<0\\ atan(sel{f}_i/othe{r}_i)& \text{if }othe{r}_i>0\\ \pi /2& \text{if }othe{r}_i=0,sel{f}_i>0\\ -\pi /2& \text{if }othe{r}_i=0,sel{f}_i<0\\ nan& \text{if }othe{r}_i=0,sel{f}_i=0\end{array}$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
self	输入	待进行atan2计算的入参，公式中的self_i，表示y坐标。shape需与other满足broadcast关系。	INT8、INT16、INT32、INT64、UINT8、BOOL、FLOAT、FLOAT16、DOUBLE、BFLOAT16	ND
other	输入	待进行atan2计算的入参，公式中的other_i，表示x坐标。shape需与self满足broadcast关系。	INT8、INT16、INT32、INT64、UINT8、BOOL、FLOAT、FLOAT16、DOUBLE、BFLOAT16	ND
out	输出	待进行atan2计算的出参，公式中的out_i。shape为self与other broadcast之后的shape。	FLOAT、FLOAT16、DOUBLE、BFLOAT16	ND

• 输入self、other支持INT8、INT16、INT32、INT64、UINT8、BOOL类型，API内部会自动将其转换为浮点类型后参与计算。
• DOUBLE通过AICPU路径计算。
• Atlas推理系列产品、Atlas训练系列产品：不支持BFLOAT16。

约束说明

• self和other的shape需满足broadcast关系，out的shape为self与other broadcast之后的shape。
• 输入输出张量的维度不超过8，支持非连续Tensor。
• 确定性计算：aclnnAtan2&aclnnInplaceAtan2默认确定性实现。

调用说明

调用方式	调用样例	说明
aclnn调用	test_aclnn_atan2	通过aclnnAtan2和aclnnInplaceAtan2接口方式调用Atan2算子。