- class torch.nn.GLU(dim=- 1)[source]¶
Applies the gated linear unit function where is the first half of the input matrices and is the second half.
dim (int) – the dimension on which to split the input. Default: -1
Input: where * means, any number of additional dimensions
>>> m = nn.GLU() >>> input = torch.randn(4, 2) >>> output = m(input)