Function rai_core::ops::gather

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