Inv

产品支持情况

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

功能说明

• 算子功能：对输入张量的每个元素逐元素计算倒数。

• 计算公式：

$$ou{t}_i=\frac{1}{sel{f}_i}$$

• 浮点（FLOAT、FLOAT16、BFLOAT16）按IEEE-754真值倒数计算（除零得 $\pm \text{Inf}$、$\pm \text{Inf}$ 得 $\pm 0$、NaN得NaN）。

• INT32按截断向零的整数倒数计算，整型下 $1/self$ 仅当 $|self|=1$ 时非零，等价为三值映射：

$$ou{t}_i=\{\begin{array}{ll}+1& sel{f}_i=+1\\ -1& sel{f}_i=-1\\ 0& \text{其他（含 }sel{f}_i=0、|sel{f}_i|\ge 2、\text{INT\_MIN}、\text{INT\_MAX)}\end{array}$$

其中 $sel{f}_i=0$ 映射为 $0$ 且不抛异常，$\text{INT\_MIN}(-2147483648)$、$\text{INT\_MAX}(2147483647)$ 均映射为 $0$ 且不溢出。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
self	输入	待进行inv计算的入参，公式中的self。支持空Tensor、非连续Tensor，shape维度0-8。	FLOAT、FLOAT16、BFLOAT16、INT32	ND
out	输出	待进行inv计算的出参，公式中的out。shape和数据类型与self一致。	FLOAT、FLOAT16、BFLOAT16、INT32	ND

约束说明

• self与out必须shape、dtype相同，逐元素计算，无广播。
• 浮点路径精度按社区标准：FLOAT rtol/atol = 1e-4，FLOAT16 = 1e-3，BFLOAT16 = 4e-3。
• INT32路径为精确匹配（bitwise，rtol = atol = 0），采用纯整型比较 + 选择实现三值映射，不经float32中转、不对INT_MIN取负，故INT_MIN/INT_MAX不溢出、self = 0不抛异常。

调用说明

调用方式	说明
图模式调用	通过算子IR构图方式调用Inv算子。样例参见test_geir_inv。