src/transformers/models/deberta_v2/modeling_deberta_v2.py/DisentangledSelfAttention/transpose_for_scores
class DisentangledSelfAttention:    def transpose_for_scores(self, x, attention_heads):
        new_x_shape = x.size()[:-1] + (attention_heads, -1)
        x = x.view(*new_x_shape)
        return x.permute(0, 2, 1, 3).contiguous().view(-1, x.size(1), x.size(-1))