AsinGrad

产品支持情况

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

功能说明

• 算子功能：AsinGrad是Asin（反正弦）算子的梯度算子，用于深度学习框架自动微分的反向传播阶段。给定正向输入y和上游梯度dy，计算输出梯度z。

• 计算公式：

  $$z_i=\frac{d{y}_i}{\sqrt{1-y_i^2}}$$

  其中：

  • y：正向Asin算子的输入tensor，值域 [-1, 1]
  • dy：上游传播的梯度（grad_output）
  • z：计算得到的输入梯度（grad_input）

参数说明

参数名	输入/输出	描述	数据类型	数据格式
y	输入	正向Asin算子的输入tensor，对应公式中的y。支持0-8维tensor，支持空tensor。shape需要与dy的shape完全相同。	BFLOAT16、FLOAT16、FLOAT32	ND
dy	输入	上游传播的梯度tensor，对应公式中的dy。支持0-8维tensor，支持空tensor。shape需要与y的shape完全相同。	BFLOAT16、FLOAT16、FLOAT32	ND
z	输出	计算得到的输入梯度tensor，对应公式中的z。shape与输入相同，数据类型与输入一致。	BFLOAT16、FLOAT16、FLOAT32	ND

约束说明

• y和dy的数据类型必须一致，z的数据类型与y/dy一致，仅支持FLOAT、FLOAT16、BFLOAT16。
• y和dy的shape必须完全相同，不支持广播。
• 支持0-8维tensor，支持空tensor（0元素）。
• y的值域应在 [-1, 1] 范围内。当 |y| = 1时，计算结果为inf；当 |y| > 1时，计算结果为NaN。
• FLOAT16走native FP16计算；BFLOAT16走BF16 → FP32中间计算；FLOAT32直接FP32计算。

调用说明

调用方式	样例代码	说明
图模式	test_geir_asin_grad.cpp	通过图模式方式调用AsinGrad算子。