bbqvec 0.0.10

Scalable, embeddable, vector storage for approximate K-nearest-neighbors (AKNN)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28

use crate::Vector;

#[inline(always)]
pub fn normalize(vec: &mut Vector) {
    let s = crate::unaligned_f32::UnalignedF32Slice::from_slice(vec.as_slice());
    let norm = crate::spaces::simple::dot_product(s, s).sqrt();
    vec.iter_mut().for_each(|v| *v /= norm);
}

#[inline(always)]
pub fn dot_product(vec: &Vector, other: &Vector) -> f32 {
    crate::spaces::simple::dot_product(vec.into(), other.into())
}

#[inline(always)]
pub fn subtract_into(vec: &mut Vector, other: &Vector) {
    for (v, o) in vec.iter_mut().zip(other.iter()) {
        *v -= o;
    }
}

#[inline(always)]
pub fn distance(vec: &Vector, other: &Vector) -> f32 {
    vec.iter()
        .zip(other.iter())
        .fold(0.0, |acc, (a, b)| acc + ((a - b) * (a - b)))
        .sqrt()
}