Example

invalid_take_along_axis_axis.kina

Expected rejection

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

Source

tensor<F32, [1, 4, 8], @host>.take_along_axis(tensor<I64, [1, 4], @host>, axis=3)