CANN/ops-math BesselI1e算子
BesselI1e产品支持情况产品是否支持Ascend 950PR/Ascend 950DT√Atlas A3 训练系列产品/Atlas A3 推理系列产品√Atlas A2 训练系列产品/Atlas A2 推理系列产品√Atlas 200I/500 A2 推理产品×Atlas 推理系列产品×Atlas 训练系列产品×功能说明算子功能计算指数缩放的第一阶修正贝塞尔函数。计算公式$$out_i e^{-|input_i|} \cdot I_1(input_i)$$其中 $I_1(x)$ 是第一阶修正贝塞尔函数。指数缩放因子 $e^{-|x|}$ 使得函数在大参数值时数值稳定。参数说明参数名输入/输出/属性描述数据类型数据格式x输入待进行BesselI1e计算的入参。FLOAT、FLOAT16、BF16NDy输出BesselI1e计算的出参。FLOAT、FLOAT16、BF16ND约束说明无调用说明调用方式调用样例说明图模式调用test_geir_bessel_i1e通过算子IR构图方式调用BesselI1e算子。【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考