當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


Python PyTorch GLU用法及代碼示例


本文簡要介紹python語言中 torch.nn.GLU 的用法。

用法:

class torch.nn.GLU(dim=- 1)

參數

dim(int) -分割輸入的維度。默認值:-1

應用門控線性單元函數 ,其中 是輸入矩陣的前半部分, 是後半部分。

形狀:
  • 輸入: 其中 * 表示任意數量的附加維度

  • 輸出: 其中

例子:

>>> m = nn.GLU()
>>> input = torch.randn(4, 2)
>>> output = m(input)

相關用法


注:本文由純淨天空篩選整理自pytorch.org大神的英文原創作品 torch.nn.GLU。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。