Function wrap_tensor

src/framework/core/module.cpp:144–158

    return static_cast<int>(rank) - 1 - logical_axis;
}

TensorValue wrap_tensor(ggml_tensor * tensor, const TensorShape & shape, ggml_type type) {
    if (tensor == nullptr) {
        throw std::runtime_error("Cannot wrap null ggml tensor");
    }

    ensure_positive_dims(shape);
    const auto ggml_dims = to_ggml_dims(shape);
    for (size_t i = 0; i < shape.rank; ++i) {
        if (tensor->ne[i] != ggml_dims[i]) {
            throw std::runtime_error("Wrapped ggml tensor shape does not match logical shape " + shape.to_string());
        }
    }

    return TensorValue{tensor, shape, type};
}

TensorValue reshape_tensor(ModuleBuildContext & ctx, const TensorValue & value, const TensorShape & new_shape) {