Skip to main content

patchify

Function patchify 

Source
pub fn patchify(
    x: &[f32],
    c: usize,
    pt: usize,
    ps: usize,
    h: usize,
    w: usize,
) -> Vec<f32>
Expand description

[T', H', W', C] patches → [n, pt·ph·pw·C] in the (pt, ph, pw, c) order the embedders were trained on.