src/transformers/models/swin2sr/modeling_swin2sr.py/Swin2SRSelfAttention/transpose_for_scores
class Swin2SRSelfAttention:    def transpose_for_scores(self, x):
        new_x_shape = x.size()[:-1] + (self.num_attention_heads, self.attention_head_size)
        x = x.view(new_x_shape)
        return x.permute(0, 2, 1, 3)