diffusion_rs_common::nn::rotary_emb

Function rope

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