将可变大小的 TensorFlow 张量转换为 std::vector(C++)
Convert TensorFlow tensor of variable size to std::vector in C++
我有一个给定的C++ tensorflow::Tensor
(可变形状(对象。
我想将其转换为std::vector
(嵌套向量,如果是多维向量(。
有没有一种干净的方法来做到这一点?
谢谢!
不是真的,不是。
你可以创建一个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::vector的OpenCL矩阵乘法
- POCO::PostgreSQL:如何将std::vector支持添加到`Binder::bind`
- std::vector的包装器,使数组的结构看起来像结构的数组
- 编译器如何区分std::vector的构造函数
- 使用 pqxx 将 std::vector 存储在 postgresql 中,并从数据库中检索它
- 在std::vector上存储带有模板的类实例
- 在main()之外初始化std::vector会导致性能下降(多线程)
- 为什么std::vector比数组慢
- std::vector::迭代器是否可以合法地作为指针
- 如何将二进制格式的 C++ 对象的 std::vector 保存到磁盘?
- 为什么std::vector和std::valarray初始化构造函数不同
- ";结果类型必须是可从输入范围的值类型""构造的;创建std::vector时
- 在没有未定义行为的情况下实现类似std::vector的容器
- 如何调整 std::vector of Eigen::MatrixXd 的大小
- 使用 std::vector::reverse_iterator 将 int 序列化为字节向量?
- 如何将AERT_Allocate与 std:vector 一起使用
- 推导 std::vector::back() 的返回类型
- 如何将原始字节附加到 std::vector?
- std::vector 没有重载函数的实例与参数列表匹配
- 如果 KEY 是 std::list 或 std::vector 而不是值,那么 std::map 的默认行为是什么?