Method forward

models/TABL/bl_layer.py:23–27

(self, x)

        self.activation = nn.ReLU()

    def forward(self, x):

        x = self.activation(self.W1 @ x @ self.W2 + self.B)

        return x