Skip to main content

gather

tenferro_tensor::cpu::indexing

Function gather 

Source 
pub fn gather(
    operand: &Tensor,
    start_indices: &Tensor,
    config: &GatherConfig,
) -> Tensor