Function kn_graph::cpu::cpu_gather

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