Ivf Pq Index Module

Rust module: cuvs::ivf_pq::index

Source: rust/cuvs/src/ivf_pq/index.rs

Index

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

Ivf-Pq ANN Index

Methods

build

1
pub fn build<T: Into<ManagedTensor>>(
2
res: &Resources,
3
params: &IndexParams,
4
dataset: T,
5
) -> Result<Index>

Builds a new Index from the dataset for efficient search.

Arguments

• res - Resources to use
• params - Parameters for building the index
• dataset - A row-major matrix on either the host or device to index

Source: rust/cuvs/src/ivf_pq/index.rs:25

new

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

Creates a new empty index

Source: rust/cuvs/src/ivf_pq/index.rs:39

search

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

Perform a Approximate Nearest Neighbors search on the Index

Arguments

• res - Resources to use
• params - Parameters to use in searching the index
• 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/ivf_pq/index.rs:56

Source: rust/cuvs/src/ivf_pq/index.rs:15