src/transformers/models/sew/modeling_sew.py/SEWAttention/_shape
class SEWAttention:    def _shape(self, tensor: torch.Tensor, seq_len: int, bsz: int):
        return tensor.view(bsz, seq_len, self.num_heads, self.head_dim).transpose(1, 2).contiguous()