src/transformers/models/cvt/modeling_tf_cvt.py/TFCvtSelfAttentionLinearProjection/call
class TFCvtSelfAttentionLinearProjection:    def call(self, hidden_state: tf.Tensor) -> tf.Tensor:
        # "batch_size, height, width, num_channels -> batch_size, (height*width), num_channels"
        batch_size, height, width, num_channels = shape_list(hidden_state)
        hidden_size = height * width
        hidden_state = tf.reshape(hidden_state, shape=(batch_size, hidden_size, num_channels))
        return hidden_state