我在Python中构建了一个模型,其中每个输入张量都具有形状(4,6,13)。我想用C ++中的模型运行预测。
我想编写一个在C ++三维数组中作为输入数据的函数:
float input[4][6][13];
并生成形状为Tensor
的{{1}}。
似乎TensorShape({4,6,13})
不支持3d数组。我看到Eigen3
可以从初始化列表初始化3d张量,但我有动态数据,而不是编译时常量。
在寻找解决方案后,我即将得出结论,我应该改变我的模型以期望一个长度为4x6x13 = 312的扁平矢量,然后重新塑造它作为第一步,但我更喜欢更清洁的解决方案。