如何在C++中将TF_Tensor转换为opencv Mat？

#include <tensorflow/c/c_api.h>
#include <opencv2/opencv.hpp>

cv::Mat TF_TensorToMat(TF_Tensor* tensor) {
    if (TF_TensorType(tensor) != TF_FLOAT) {
        throw std::runtime_error("Tensor type must be float");
    }
    
    if (TF_NumDims(tensor) != 4) {
        throw std::runtime_error("Tensor must be 4-dimensional");
    }
    
    int64_t batch = TF_Dim(tensor, 0);
    int64_t height = TF_Dim(tensor, 1);
    int64_t width = TF_Dim(tensor, 2);
    int64_t channels = TF_Dim(tensor, 3);
    
    float* data = static_cast<float*>(TF_TensorData(tensor));
    
    cv::Mat mat(height, width, CV_32FC(channels));
    
    for (int64_t b = 0; b < batch; b++) {
        for (int64_t h = 0; h < height; h++) {
            for (int64_t w = 0; w < width; w++) {
                for (int64_t c = 0; c < channels; c++) {
                    float value = data[b * height * width * channels + h * width * channels + w * channels + c];
                    mat.at<cv::Vec3f>(h, w)[c] = value;
                }
            }
        }
    }
    
    return mat;
}

TF_Tensor* tensor = ...;  // 你的TF_Tensor对象

cv::Mat mat = TF_TensorToMat(tensor);