QKV Projections And Tensor Shapes