torch.nn.functional.gelu

torch.nn.functional.gelu(input, approximate='none') → Tensor

當 approximate 引數為 'none' 時，按元素應用以下函式 $\text{GELU}(x) = x * \Phi(x)$

其中 $\Phi(x)$ 是高斯分佈的累積分佈函式。

當 approximate 引數為 'tanh' 時，Gelu 透過以下公式近似計算：

$$\text{GELU}(x) = 0.5 * x * (1 + \text{Tanh}(\sqrt{2 / \pi} * (x + 0.044715 * x^3)))$$

參見 高斯誤差線性單元 (GELUs)。