Function candle_nn::rotary_emb::rope_thd

pub fn rope_thd(xs: &Tensor, cos: &Tensor, sin: &Tensor) -> Result<Tensor>