Skip to main content

build_rope_tables

Function build_rope_tables 

Source
pub fn build_rope_tables(
    inv_freq: &[f32],
    head_dim: usize,
    max_seq: usize,
    device: &Device,
) -> Result<(Tensor<2>, Tensor<2>)>
Expand description

Build cos/sin tables from explicit inv_freq ([half]).