GELU

class GELU(name=None)[源代码]

对每个元素应用函数:

\[\text{GELU}(x) = x\Phi(x)\]

\(\Phi(x)\) 是高斯分布的累积分布函数 。

实际案例

>>> import numpy as np
>>> data = mge.tensor(np.array([-2,-1,0,1,2,]).astype(np.float32))
>>> gelu = M.GELU()
>>> output = gelu(data)
>>> with np.printoptions(precision=4):
...     print(output.numpy())
[-0.0455 -0.1587  0.      0.8413  1.9545]