Example

invalid_take_along_axis_device.kina

Expected rejection

This program is intentionally invalid. The compiler should reject its type or shape.

Source

tensor<F32, [2, 3], @device(0)>.take_along_axis(tensor<I64, [2], @device(0)>, axis=1)