Skip to main content

float_select

ruprim_host::gather_scatter::dispatch_float

Function float_select 

Source 
pub fn float_select(
    tensor: HostTensor,
    dim: usize,
    indices: HostTensor,
) -> HostTensor