src/transformers/models/swin/modeling_tf_swin.py/TFSwinSelfAttention/transpose_for_scores
class TFSwinSelfAttention:    def transpose_for_scores(self, x: tf.Tensor) -> tf.Tensor:
        new_x_shape = shape_list(x)[:-1] + [self.num_attention_heads, self.attention_head_size]
        x = tf.reshape(x, new_x_shape)
        return tf.transpose(x, (0, 2, 1, 3))