mindspore.ops.GLU
- class mindspore.ops.GLU(axis=- 1)[源代码]
门线性单元函数(Gated Linear Unit function)。
警告
这是一个实验性API,后续可能修改或删除。
更多参考详见
mindspore.ops.glu()
。- 参数:
axis (int,可选) - 指定分割轴。是一个在范围[-rank(x), rank(x))内的整数。默认值:-1,输入 x 的最后一维。
- 输入:
x (Tensor) - 输入Tensor, x.shape[axis] 必须为偶数。
- 输出:
Tensor,数据类型与输入 x 相同。
- 支持平台:
Ascend
CPU
样例:
>>> from mindspore import ops, Tensor >>> from mindspore import dtype as mstype >>> import numpy as np >>> axis = 0 >>> x = Tensor(np.array([0.3220, 0.9545, 0.7879, 0.0975, 0.3698, ... 0.5135, 0.5740, 0.3435, 0.1895, 0.8764, ... 0.4980, 0.9673, 0.9879, 0.6988, 0.9022, ... 0.9304, 0.1558, 0.0153, 0.1559, 0.9852]).reshape([2, 2, 5]), mstype.float32) >>> glu = ops.GLU(axis=axis) >>> y = glu(x) >>> print(y) [[[0.20028052 0.6916126 0.57412136 0.06512236 0.26307625] [0.3682598 0.3093122 0.17306386 0.10212085 0.63814086]]]