mindspore.ops.erfinv

查看源文件
mindspore.ops.erfinv(input)[源代码]

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

\[erfinv(erf(x)) = x\]

其中 \(x\) 代表输入Tensor。

参数:
  • input (Tensor) - 输入Tensor。支持数据类型:

    • Ascend: float16、float32。

    • GPU/CPU: float16、float32、float64。

返回:

Tensor,具有与 input 相同的数据类型和shape。

异常:
  • TypeError - input 的数据类型不是float16、float32、float64。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> x = Tensor(np.array([0, 0.5, -0.9]), mindspore.float32)
>>> output = ops.erfinv(x)
>>> print(output)
[ 0.          0.47695306 -1.1630805 ]