def forward(self, x):
        """
        Pass the input through the BIOT encoder, and then through the
        classification head.

        Parameters
        ----------
        x: Tensor
            (batch_size, n_channels, n_times)

        Returns
        -------
        out: Tensor
            (batch_size, n_outputs)
        (out, emb): tuple Tensor
            (batch_size, n_outputs), (batch_size, emb_size)
        """
        emb = self.encoder(x)
        x = self.final_layer(emb)

        if self.return_feature:
            return x, emb
        else:
            return x

        Pass the input through the BIOT encoder, and then through the
        classification head.

        Parameters
        ----------
        x: Tensor
            (batch_size, n_channels, n_times)

        Returns
        -------
        out: Tensor
            (batch_size, n_outputs)
        (out, emb): tuple Tensor
            (batch_size, n_outputs), (batch_size, emb_size)
        
braindecode/models/biot.py