Function cpu_gather

Source
pub fn cpu_gather<T: Clone>(
    input: &Tensor<T>,
    axis: usize,
    indices: DTensor,
) -> Tensor<T>