Python PyTorch erfinv用法及代码示例

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

用法:

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

参数：

input(Tensor) -输入张量。

关键字参数：

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

计算 input 的逆误差函数。逆误差函数在(-1, 1) 范围内定义为：

\mathrm{erfinv}(\mathrm{erf}(x)) = x

例子：

>>> torch.special.erfinv(torch.tensor([0, 0.5, -1.]))
tensor([ 0.0000,  0.4769,    -inf])