快捷方式

torch.nn.functional.glu

torch.nn.functional.glu(input, dim=-1) Tensor[源代码]

门控线性单元。计算

GLU(a,b)=aσ(b)\text{GLU}(a, b) = a \otimes \sigma(b)

其中 input 沿 dim 一分为二,形成 abσ\sigma 是 sigmoid 函数,\otimes 是矩阵之间的逐元素乘积。

参见 使用门控卷积网络进行语言建模

参数
  • input (张量) – 输入张量

  • dim (int) – 用于分割输入的维度。默认值:-1

返回类型

张量

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源