Function npyiter_get_op_axis

numpy/core/src/multiarray/nditer_constr.c:1467–1478

* Removes the (additive) NPY_ITER_REDUCTION_AXIS indication and sets * is_forced_broadcast to 1 if it is set. Otherwise to 0. * * @param axis The op_axes[i] to normalize. * @param reduction_axis Output 1 if a reduction axis, otherwise 0. * @returns The normalized axis (without reduce axis flag). */

 * @returns The normalized axis (without reduce axis flag).
 */
static inline int
npyiter_get_op_axis(int axis, npy_bool *reduction_axis) {
    npy_bool forced_broadcast = axis >= NPY_ITER_REDUCTION_AXIS(-1);

    if (reduction_axis != NULL) {
        *reduction_axis = forced_broadcast;
    }
    if (forced_broadcast) {
        return axis - NPY_ITER_REDUCTION_AXIS(0);
    }
    return axis;
}

/*
 * Fills in the AXISDATA for the 'nop' operands, broadcasting