modify glu ops doc

This commit is contained in:
lianliguang 2022-10-27 09:56:19 +08:00
parent 4c0408d22e
commit c5931973a3
1 changed files with 2 additions and 1 deletions

View File

@ -10,12 +10,13 @@ mindspore.nn.GLU
其中,:math:`a` 表示输入Tensor的前一半元素:math:`b` 表示输入Tensor的另一半元素。
这里 :math:`\sigma` 为sigmoid函数:math:`*` 为矩阵的基本乘。
参数:
- **axis** (int) - 指定分割轴。数据类型为整型,默认值:-1。
输入:
- **x** (Tensor) - Tensor的shape为 :math:`(\ast_1, N, \ast_2)``x` 必须在 `axis` 轴能够被平均分成两份。
- **x** (Tensor) - Tensor的shape为 :math:`(\ast_1, N, \ast_2)`
输出:
Tensor数据类型与输入 `x` 相同shape等于 `x` 按照 `axis` 拆分后的一半。