forked from mindspore-Ecosystem/mindspore
modify glu ops doc
This commit is contained in:
parent
4c0408d22e
commit
c5931973a3
|
@ -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` 拆分后的一半。
|
||||
|
|
Loading…
Reference in New Issue