ferritin_plms::featurize::utilities

Function topk_last_dim

Source 
pub fn topk_last_dim(xs: &Tensor, topk: usize) -> Result<(Tensor, Tensor)>