Function gather

pub fn gather(x: &Tensor, dim: impl Dim, index: &Tensor) -> Tensor