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])