是否有一种干净的方法来创建用于在C ++中提供Tensorflow预测的3d张量?

时间:2017-12-16 22:40:58

标签: c++ tensorflow

我在Python中构建了一个模型,其中每个输入张量都具有形状(4,6,13)。我想用C ++中的模型运行预测。

我想编写一个在C ++三维数组中作为输入数据的函数:

float input[4][6][13];

并生成形状为Tensor的{​​{1}}。

似乎TensorShape({4,6,13})不支持3d数组。我看到Eigen3可以从初始化列表初始化3d张量,但我有动态数据,而不是编译时常量。

在寻找解决方案后,我即将得出结论,我应该改变我的模型以期望一个长度为4x6x13 = 312的扁平矢量,然后重新塑造它作为第一步,但我更喜欢更清洁的解决方案。

0 个答案:

没有答案