Kina.ai

Example

gather_2d.kina

Source
tensor<F32, [10, 20], @host>.gather(tensor<I64, [4, 5], @host>)