Function reduce_axis

src/framework/modules/primitive_modules.cpp:139–163

    if (rank == 0) {
        throw std::runtime_error("Reduce input rank must be positive");
    }
    const int normalized = axis < 0 ? static_cast<int>(rank) + axis : axis;
    if (normalized < 0 || normalized >= static_cast<int>(rank)) {
        throw std::runtime_error("Reduce axis out of range");
    }
    return normalized;
}

core::TensorShape reduced_shape(const core::TensorShape & shape, int axis) {
    core::TensorShape out = shape;
    out.dims[static_cast<size_t>(axis)] = 1;
    return out;
}

template <typename Fn>
core::TensorValue reduce_axis(
    core::ModuleBuildContext & ctx,
    const core::TensorValue & input,
    const ReduceConfig & config,
    Fn fn) {
    if (ctx.ggml == nullptr) {
        throw std::runtime_error("ModuleBuildContext.ggml is null");
    }
    core::validate_rank_between(input, 1, core::kMaxTensorRank, "input");
    const int axis = normalize_reduce_axis(config.axis, input.shape.rank);
    if (axis == static_cast<int>(input.shape.rank - 1)) {
        const auto contiguous = tensor_layout::ensure_contiguous_layout_if_needed(ctx, input);
        return core::wrap_tensor(fn(ctx.ggml, contiguous.tensor), reduced_shape(input.shape, axis), GGML_TYPE_F32);