src/transformers/models/mobilevit/modeling_tf_mobilevit.py/TFMobileViTSelfAttention/transpose_for_scores
class TFMobileViTSelfAttention:    def transpose_for_scores(self, x: tf.Tensor) -> tf.Tensor:
        batch_size = tf.shape(x)[0]
        x = tf.reshape(x, shape=(batch_size, -1, self.num_attention_heads, self.attention_head_size))
        return tf.transpose(x, perm=[0, 2, 1, 3])