在 C++ 中将可变大小的 TensorFlow 张量转换为 std::vector答案

【问题标题】：Convert TensorFlow tensor of variable size to std::vector in C++在 C++ 中将可变大小的 TensorFlow 张量转换为 std::vector
【发布时间】：2017-06-30 09:39:53
【问题描述】：

我有一个给定的 C++ tensorflow::Tensor（可变形状）对象。

我想将其转换为std::vector（嵌套向量，如果是多维向量）。

有没有干净的方法来做到这一点？

谢谢！

【问题讨论】：

你想用这个向量实现什么？你知道张量有多少维吗？否则，如果不使用某种多态性，您甚至无法命名向量的类型，这会使一切变得复杂。

标签： c++ tensorflow

【解决方案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
}

【讨论】：

好吧，我需要将张量插入另一个与 std 嵌套向量一起使用的代码中......如果我有一个简单的迭代器，那么我需要先验知道维数以保持我的维数假设。
你有void doStuff(vector<vector<vector<int>>> & input)这样的功能吗？您可以创建一个函数，将特定形状的张量转换为该形状的向量，但这并不通用
是的，你是对的。对于迭代器，for (auto it = std::begin(Y_pred); it!=std::end(Y_pred); ++it) { ... } 对我不起作用。 Y_pred 是一个张量流张量。