_get_scalar_func_loopy_program/loopy transform logic: Allow any number of axes in array