Skip to main content

float_argtopk

ruprim_host::sort::dispatch

Function float_argtopk 

Source 
pub fn float_argtopk(
    tensor: HostTensor,
    dim: usize,
    k: usize,
    out_dtype: IntDType,
) -> HostTensor