forked from mindspore-Ecosystem/mindspore
19 lines
672 B
ReStructuredText
19 lines
672 B
ReStructuredText
|
mindspore.ops.glu
|
|||
|
|
=================
|
|||
|
|
|
|||
|
|
.. py:class:: mindspore.ops.glu(input_x, axis=-1)
|
|||
|
|
|
|||
|
|
门线性单元函数（Gated Linear Unit function）。
|
|||
|
|
|
|||
|
|
.. math:: {GLU}(a, b)= a \otimes \sigma(b)`
|
|||
|
|
|
|||
|
|
|
|||
|
|
其中，:math:`a` 表示输入input_x 拆分后 Tensor的前一半元素，:math:`b` 表示输入拆分Tensor的另一半元素。
|
|||
|
|
|
|||
|
|
参数：
|
|||
|
|
- **axis** (`int`) - 指定分割轴。数据类型为整型，默认值：0。
|
|||
|
|
- **x** (`Tensor`) - Tensor的shape为 (x_1, x_2, ..., x_R) 。x 必须在axis 轴能够被平均分成两份。
|
|||
|
|
|
|||
|
|
返回：
|
|||
|
|
Tensor，数据类型与输入 x 相同，shape等于 x 按照axis 拆分后的一半。
|