将可变大小的 TensorFlow 张量转换为 std::vector(C++)

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
}