You can not select more than 25 topics
Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
|
- mindspore.ops.GeLU
- ==================
-
- .. py:class:: mindspore.ops.GeLU()
-
- 高斯误差线性单元激活函数(Gaussian Error Linear Units activation function)。
-
- 在 `Gaussian Error Linear Units (GELUs) <https://arxiv.org/abs/1606.08415>`_ 文章中对GeLU函数进行了介绍。
- 此外,也可以参考 `BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding <https://arxiv.org/abs/1810.04805>`_ 。
-
- GeLU函数定义如下:
-
- .. math::
- GELU(x_i) = x_i*P(X < x_i)
-
- 其中 :math:`P` 是标准高斯分布的累积分布函数, :math:`x_i` 是输入的元素。
-
- **输入:**
-
- - **x** (Tensor) - 激活函数GeLU的输入,数据类型为float16或float32。
-
- **输出:**
-
- Tensor,数据类型和shape与 `x` 的相同。
-
- **异常:**
-
- - **TypeError** - `x` 不是Tensor。
- - **TypeError** - `x` 的数据类型既不是float16也不是float32。
|