src/transformers/models/cvt/modeling_cvt.py/CvtSelfAttentionLinearProjection/forward
class CvtSelfAttentionLinearProjection:    def forward(self, hidden_state):
        batch_size, num_channels, height, width = hidden_state.shape
        hidden_size = height * width
        # rearrange " b c h w -> b (h w) c"
        hidden_state = hidden_state.view(batch_size, num_channels, hidden_size).permute(0, 2, 1)
        return hidden_state