Python PyTorch ndtri用法及代码示例

本文简要介绍python语言中 torch.special.ndtri 的用法。

用法:

torch.special.ndtri(input, *, out=None) → Tensor

参数：

input(Tensor) -输入张量。

关键字参数：

out(Tensor,可选的) -输出张量。

计算自变量 x，其高斯概率密度函数(从负无穷到 x 积分)下的面积等于 input ，按元素计算。

\text{ndtri}(p) = \sqrt{2}\text{erf}^{-1}(2p - 1)

注意

也称为正态分布的分位数函数。

例子：

>>> torch.special.ndtri(torch.tensor([0, 0.25, 0.5, 0.75, 1]))
tensor([   -inf, -0.6745,  0.0000,  0.6745,     inf])