torch - 转换 C++ 向量> to torch::tensor

Question

我正在尝试将我的 c++ 向量转换为火炬张量。但是，我的代码返回不正确的转换。

    cout << history << endl;
    auto options1 = torch::TensorOptions().dtype(torch::kFloat32);
    input = torch::from_blob(history.data(), {size, 1, 6}, options1).to(torch::kFloat32);

    cout << input << endl;

上面的代码返回以下输出：

-9 -3 -3 -12 -0 -0 -12 -2 -3 -12 -0 -0

(1,.,.) = 
 -9.8681e-32  4.5793e-41 -9.8682e-32  4.5793e-41 -9.8682e-32  4.5793e-41




(2,.,.) = 
 -9.8682e-32  4.5793e-41 -9.8683e-32  4.5793e-41 -9.8683e-32  4.5793e-41




[ CPUFloatType{2,1,6} ]

Answer 1

您不能使用该方法将 2D 向量或更多维度转换为 Tensor from_blob()，但您可以使用此方法来克服该问题：

vector<float> linearize(const vector<vector<float>>& vec_vec) {
    vector<float> vec;
    for (const auto& v : vec_vec) {
        for (auto d : v) {
            vec.push_back(d);
        }
    }
    return vec;
}

并且，通过这种方式，您可以将其转换为张量（m，n 向量的宽度和高度）：

vector<float> vec = linearize(your2Dvector);
torch::Tensor t = torch::from_blob(vec.data(), {n,m});