Function rotate_blocks_two_sided_partial 

pub fn rotate_blocks_two_sided_partial<const D: usize, const DB: usize>(
    v: Tensor<D>,
    ql: Tensor<DB>,
    qr: Tensor<DB>,
    rope_width: usize,
) -> Tensor<D>

Two-sided counterpart of rotate_blocks_partial: rotates the first rope_width entries of the state_rank axis by v ↦ ql ⊗ v ⊗ qr, passing the remainder through.