Method __init__

ML/src/python/neuralforge/nn/activations.py:51–53

(self, dim=-1)

class GLU(nn.Module):
    def __init__(self, dim=-1):
        super().__init__()
        self.dim = dim
    
    def forward(self, x):
        a, b = x.chunk(2, dim=self.dim)