Function reshape_tensor

src/framework/core/module.cpp:160–191

}

TensorValue reshape_tensor(ModuleBuildContext & ctx, const TensorValue & value, const TensorShape & new_shape) {
    if (!value.valid()) {
        throw std::runtime_error("Cannot reshape an invalid tensor");
    }
    if (value.shape.num_elements() != new_shape.num_elements()) {
        throw std::runtime_error("Reshape element count mismatch");
    }
    if (ctx.ggml == nullptr) {
        throw std::runtime_error("ModuleBuildContext.ggml is null");
    }

    const auto ggml_dims = to_ggml_dims(new_shape);
    ggml_tensor * reshaped = nullptr;
    switch (new_shape.rank) {
        case 1:
            reshaped = ggml_reshape_1d(ctx.ggml, value.tensor, ggml_dims[0]);
            break;
        case 2:
            reshaped = ggml_reshape_2d(ctx.ggml, value.tensor, ggml_dims[0], ggml_dims[1]);
            break;
        case 3:
            reshaped = ggml_reshape_3d(ctx.ggml, value.tensor, ggml_dims[0], ggml_dims[1], ggml_dims[2]);
            break;
        case 4:
            reshaped = ggml_reshape_4d(ctx.ggml, value.tensor, ggml_dims[0], ggml_dims[1], ggml_dims[2], ggml_dims[3]);
            break;
        default:
            throw std::runtime_error("Unsupported reshape rank");
    }

    return wrap_tensor(reshaped, new_shape, value.type);
}

bool has_backend_addressable_layout(const ggml_tensor * tensor) noexcept {
    return tensor != nullptr &&