Example

gather_2d.kina

Source

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