如何在Tensorflow的c ++ API中使用复杂数据类型？

debugcn 发表于 Dev

xor_girl

我已经使用Tensorflow C ++ API有一段时间了，我有一个共享的构建工作，所以我可以加载在C ++ API的曲线图，并执行类似的示出简单的例子在这里。我需要在C ++中使用Tensorflow的Complex Data类型，但是我找不到关于它的任何示例。您能推荐一个非常简单的复杂数据类型用法吗？

阿拉法特

如果有人否决了您，那么对不起，新手。对于新用户来说，堆栈溢出可能有点怪异，但是对于程序员来说，这是个好地方，因此您一定要尽其所能使用它。

至于您的问题，您只需使用复数库即可。

如果您正在使用共享版本，则可以尝试以下示例（在为所需的必要文件调整了包含路径之后）

#include <bits/stdc++.h>
#include "tensorflow/core/public/tensor_c_api.h"
#include "tf_session_helper.h"
#include "tf_session_helper.cc"
#include "tf_tensor_helper.cc"
using namespace std;

main()
{ 

    long long adims[] ={3};
    std::complex<double> aData[3]={{1, 2},{3, 4}, {231,452}};     
    auto c = tensorflow::TF_NewTensor_wrapper(TF_DataType::TF_COMPLEX128,adims,(sizeof(adims)/sizeof(*adims)),aData,16*(sizeof(aData)/sizeof(*aData)));
    std::complex<double>* tensor_data = static_cast<std::complex<double>*>(TF_TensorData(c));
    auto dims = TF_NumDims(c);
    auto total_elements = 1;
    for (int i = 0; i < dims; ++i) {
        total_elements *= TF_Dim(c, i);
    }
    for (int i = 0; i < total_elements; ++i) {
        cout << std::real(tensor_data[i]) << " " << std::imag(tensor_data[i]) << endl; 
    }

}

如果您有任何疑问，请告诉我。

本文收集自互联网，转载请注明来源。

如有侵权，请联系[email protected] 删除。