src/transformers/models/yoso/modeling_yoso.py/YosoSelfAttention/transpose_for_scores
class YosoSelfAttention:    def transpose_for_scores(self, layer):
        new_layer_shape = layer.size()[:-1] + (self.num_attention_heads, self.attention_head_size)
        layer = layer.view(*new_layer_shape)
        return layer.permute(0, 2, 1, 3)