Function tch::vision::imagenet::top

source · 
pub fn top(tensor: &Tensor, k: i64) -> Vec<(f64, String)>
Expand description

Returns the top k classes as well as the associated scores.