def forward(self, x):
        """
        Apply the convolution to the input tensor.
        then merge the output tensor to the desired shape.

        Parameters:
        -----------
        x: torch.Tensor
            Input tensor of shape (Batch, Channels, n_patchs, patch_size).

        Return:
        -------
        x: torch.Tensor
            Output tensor of shape (Batch, n_patchs, patch_size, channels).
        """
        x = self.proj(x)
        x = self.merge_transpose(x)
        return x

        Apply the convolution to the input tensor.
        then merge the output tensor to the desired shape.

        Parameters:
        -----------
        x: torch.Tensor
            Input tensor of shape (Batch, Channels, n_patchs, patch_size).

        Return:
        -------
        x: torch.Tensor
            Output tensor of shape (Batch, n_patchs, patch_size, channels).
        
braindecode/models/labram.py