Как использовать сложный тип данных в C++ API Tensorflow?

Question

Я использую Tensorflow C++ API в течение некоторого времени, и у меня есть совместная сборка, поэтому я могу загрузить график на C++ api и выполнить простые примеры, подобные показанному here. Мне нужно использовать сложный тип данных тензорного потока в C++, но на этом я не могу найти никакого примера. Можете ли вы порекомендовать очень простое использование сложного типа данных?

Answer 1

Если кто-то проголосовал за вас, то я извиняюсь, новичок. Переполнение стека может быть немного странным для новых пользователей, но это отличное место для программистов, и вы должны обязательно использовать его в меру своих возможностей.

Что касается вас, вы можете просто использовать комплексные номера library.

Если у вас есть общая сборка работает, то вы можете попробовать этот пример (после регулировки включают пути для необходимых файлов, необходимых)

#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; 
    } 

} 

Позвольте мне знать, если у вас есть какие-либо вопросы.