可変サイズのTensorFlowテンソルをC++のstd :: vectorに変換します

Question

私は与えられたC++ tensorflow::Tensor（可変形状）オブジェクトを持っています。

std::vector（多次元ベクトルの場合は入れ子ベクトル）に変換したいと思います。

これを行うにはきれいな方法がありますか？

ありがとうございます！

Answer 1

実際はありません。

vector<boost::any>を作成してネストすることができますが、消費するコードは面倒です。

あなたは上の操作されている形状どのようなコンパイル時にわかっている場合、あなたは、このベクターを用いて達成するために何をしようとしている

temlate<size_t N, typename T> 
struct tensor_traits { 
    using type = std::vector<tensor_traits<N-1, T>::type>; 
} 

template<typename T> 
struct tensor_traits<0, T> { 
    using type = T; 
} 

template<size_t N, typename T> 
tensor_traits<N, T>::type toVector(const tensorflow::Tensor & input) 
{ 
    assert(/* shape and type match */); 
    tensor_traits<N, T>::type result; 
    for(size_t i = 0; i < input.dim_size(0); ++i) 
    { 
     result.push_back(toVector<N-1, T>(input[i])); 
     // Or however you get a smaller Tensor from a larger one. The documentation is unclear 
    } 
    return result; 
} 

template<typename T> 
T toVector<0, T>(const T & input) 
{ 
    return input; 
    // Base case of single value 
}