c++ - 如何在 C++ 中制作 tensorflow oneHot？

Question

在我的图像检测 python 代码中，我正在使用这个

xlabel = tf.one_hot(1, nClass).eval()

现在我正在使用 C++ 让模型在 iOS 上运行。任何人都可以给我可以做同样事情的 c++ 代码片段吗？我的目标是获得大小为 60 的 oneHot 对象。开启值为 1，关闭值为 0。

Answer 1

我无法通过 Tensorflow C++ 直接 API 执行此操作，但我找到了解决方法。如果它可以帮助像我这样的其他人，那就是:

std::vector<float> oneHotVec;
for(int i = 0; i < 62 ; i++)
{
    if (i == question)
        oneHotVec.push_back(1.0f);
    else
        oneHotVec.push_back(0.0f);
}
tensorflow::Tensor z(tensorflow::DT_FLOAT, tensorflow::TensorShape({1, 62}));
for (auto j = 0; j < 62; j++)
{ 
    z.matrix<float>().operator()(0, j) = oneHotVec.at(j);
}

这里“z”是张量需要的一个热 vector 。