Brute Force Module

Rust module: cuvs::brute_force

Source: rust/cuvs/src/brute_force.rs

Brute Force KNN

Index

1
#[derive(Debug)]
2
pub struct Index {
3
    /* private fields */
4
}

Brute Force KNN Index

Methods

build

1
pub fn build<T: Into<ManagedTensor>>(
2
res: &Resources,
3
metric: DistanceType,
4
metric_arg: Option<f32>,
5
dataset: T,
6
) -> Result<Index>

Builds a new Brute Force KNN Index from the dataset for efficient search.

Arguments

• res - Resources to use
• metric - DistanceType to use for building the index
• metric_arg - Optional value of p for Minkowski distances
• dataset - A row-major matrix on either the host or device to index

Source: rust/cuvs/src/brute_force.rs:32

new

1
pub fn new() -> Result<Index>

Creates a new empty index

Source: rust/cuvs/src/brute_force.rs:54

search

1
pub fn search(
2
&self,
3
res: &Resources,
4
queries: &ManagedTensor,
5
neighbors: &ManagedTensor,
6
distances: &ManagedTensor,
7
) -> Result<()>

Perform a Nearest Neighbors search on the Index

Arguments

• res - Resources to use
• queries - A matrix in device memory to query for
• neighbors - Matrix in device memory that receives the indices of the nearest neighbors
• distances - Matrix in device memory that receives the distances of the nearest neighbors

Source: rust/cuvs/src/brute_force.rs:70

Source: rust/cuvs/src/brute_force.rs:16