代码拉取完成,页面将自动刷新
.. py:class:: mindspore.nn.GLU(axis=-1) 门线性单元函数(Gated Linear Unit function)。 .. math:: {GLU}(a, b)= a \otimes \sigma(b) 其中,:math:`a` 表示输入Tensor的前一半元素,:math:`b` 表示输入Tensor的另一半元素。 这里 :math:`\sigma` 为sigmoid函数,:math:`\otimes` 是Hadamard乘积。 参数: - **axis** (int) - 指定分割轴。数据类型为整型,默认值: ``-1`` ,输入x的最后一维。 输入: - **x** (Tensor) - Tensor的shape为 :math:`(\ast_1, N, \ast_2)` 。 `*` 表示任意数量的维度。 输出: Tensor,数据类型与输入 `x` 相同,shape为 :math:`(\ast_1, M, \ast_2)`,其中 :math:`M=N/2` 。
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。