def forward(self, x):
        """
        x shape = [b, h, w, c]
        Returns tuple with x, with new shape = [b, seq_len, c_out]
        """

        x = rearrange(x, 'b h w c -> b c h w')
        x = self.encoder(x)
        x = rearrange(x, 'b c h w -> b (h w) c')
        return x

        x shape = [b, h, w, c]
        Returns tuple with x, with new shape = [b, seq_len, c_out]
        
analyze/convnexts4nd/src/tasks/encoders.py