文件最后提交记录最后更新时间
2 个月前
2 个月前
2 个月前
2 个月前
2 个月前
2 个月前
2 个月前
2 个月前
README

AcoshGrad

产品支持情况

产品 是否支持
Atlas A2 训练系列产品/Atlas A2 推理系列产品 √
Atlas A3 训练系列产品/Atlas A3 推理系列产品 √

功能说明

  • 算子功能:AcoshGrad 是 Acosh 算子的反向梯度算子。输入前向 Acosh 的输出 y 和上游梯度 dy,输出对前向输入的梯度 dx。

  • 计算公式:

    dxi=dyisinh⁡(yi)dx_i = \frac{dy_i}{\sinh(y_i)}

    当前 AscendC 实现保留 TBE DSL 路径,使用 y/8 上的 Taylor 多项式和 3 次 sqrt 倍角公式近似计算 sinh(y),再执行 dy / sinh(y)。

参数说明

参数名 输入/输出 描述 数据类型 数据格式
y 输入 前向 Acosh 算子的输出 tensor。 BFLOAT16、FLOAT16、FLOAT32 ND
dy 输入 上游传播的梯度 tensor,shape 和 dtype 必须与 y 相同。 BFLOAT16、FLOAT16、FLOAT32 ND
dx 输出 反向梯度输出 tensor,shape 和 dtype 与 y 相同。 BFLOAT16、FLOAT16、FLOAT32 ND

约束说明

  • y 和 dy 的数据类型必须一致,仅支持 FLOAT16、FLOAT32、BFLOAT16。
  • y 和 dy 的 shape 必须完全相同,不支持 broadcast。
  • dx 的 shape 和 dtype 与 y 保持一致。
  • 支持 ND 格式,元素数需大于 0。
  • sinh(y)==0 时结果按硬件除法行为产生 inf 或 nan,接口层不做额外截断。
  • FLOAT16 和 BFLOAT16 输入在 kernel 内部转换为 FLOAT32 计算,输出前 cast 回原 dtype。

调用说明

调用方式 样例代码 说明
aclnn 接口 test_aclnn_acosh_grad.cpp 通过 aclnnAcoshGrad 接口方式调用 AcoshGrad 算子。

贡献说明

贡献者 贡献方 贡献算子 贡献时间 贡献内容
GMOW 西北工业大学智能感知交互实验室 AcoshGrad 2026/7/25 AcoshGrad 算子 AscendC 实现