Skip to main content

unsplit_head

Function unsplit_head 

Source
pub fn unsplit_head(
    d: &[f32],
    t: usize,
    c: usize,
    h: usize,
    which: usize,
) -> Vec<f32>
Expand description

split_heads backward for one of q/k/v: place d [h, t, hd] into the which third of a zeroed [t, 3c] buffer.