ferritin_plms::featurize::utilities

Function get_nearest_neighbours

Source 
pub fn get_nearest_neighbours(
    cb: &Tensor,
    mask: &Tensor,
    y: &Tensor,
    y_t: &Tensor,
    y_m: &Tensor,
    number_of_ligand_atoms: i64,
) -> Result<(Tensor, Tensor, Tensor, Tensor)>