評價此頁

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 是矩陣之間的逐元素乘積。

請參見 Language Modeling with Gated Convolutional Networks

引數
  • input (Tensor) – 輸入張量

  • dim (int) – 用於分割輸入的維度。預設值: -1

返回型別

張量