Ndtri

产品支持情况

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

功能说明

• 算子功能：逆标准正态累积分布函数（probit / inverse normal CDF）。对输入概率张量逐元素计算标准正态分位点，对齐 PyTorch torch.special.ndtri 与 SciPy scipy.special.ndtri 语义。底层采用 Cephes 数学库的分区间有理逼近算法。

• 计算公式：

$$y_i=\mathrm{n}\mathrm{d}\mathrm{t}\mathrm{r}\mathrm{i}(x_i)={\Phi }^{-1}(x_i)=\sqrt{2}\cdot {\mathrm{e}\mathrm{r}\mathrm{f}}^{-1}(2x_i-1),x_i\in (0,1)$$

• 特殊值处理（与 PyTorch / SciPy 一致）：

输入 $x_i$	输出 $y_i$
0	-inf
1	+inf
$x_i<0$ 或 $x_i>1$	NaN
NaN / +inf / -inf	NaN

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	待进行 ndtri 计算的概率张量，公式中的 x_i，推荐值域 (0, 1)。	FLOAT、FLOAT16、BFLOAT16	ND
y	输出	逆标准正态 CDF 的计算结果，公式中的 y_i。	FLOAT、FLOAT16、BFLOAT16	ND

约束说明

无

调用说明

调用方式	调用样例	说明
图模式调用	test_geir_ndtri	通过算子IR构图方式调用 Ndtri 算子。