View source: R/nnf-activation.R
| nnf_glu | R Documentation |
The gated linear unit. Computes:
nnf_glu(input, dim = -1)
input |
(Tensor) input tensor |
dim |
(int) dimension on which to split the input. Default: -1 |
GLU(a, b) = a \otimes \sigma(b)
where input is split in half along dim to form a and b, \sigma
is the sigmoid function and \otimes is the element-wise product
between matrices.
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.