Example

invalid_log_softmax_axis.kina

Expected rejection

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

Source

tensor<F32, [2, 3], @host>.log_softmax(axis=2)