tch::vision::imagenet
pub fn top(tensor: &Tensor, k: i64) -> Vec<(f64, String)>
Returns the top k classes as well as the associated scores.