GELU

class torch.nn.GELU(approximate='none')

应用 Gaussian Error Linear Units 函数。

$$\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)))$$

参数

approximate (str, 可选) – 要使用的 gelu 近似算法：'none' | 'tanh'。默认值：'none'

形状

• 输入： $(*)$，其中 $*$ 表示任意数量的维度。

• 输出： $(*)$，与输入形状相同。

示例

>>> m = nn.GELU()
>>> input = torch.randn(2)
>>> output = m(input)