Function compute_nearest_neighbors

pub fn compute_nearest_neighbors(
    coords: &Tensor,
    mask: &Tensor,
    k: usize,
    eps: f32,
) -> Result<(Tensor, Tensor)>

Retrieve the nearest Neighbor of a set of coordinates. Usually used for CA carbon distance.