在C ++中将可变大小的TensorFlow张量转换为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
}