如何在libtorch中将2d张量修改为3d？

Question

我有一个2d数组向量，我已经将它转换为张量，但是如何修改张量的维度，我想将维度从2d修改为3d？

std::vector<std::vector<float>> voice(434, std::vector<float>(80))
ifstream fp("data.txt");
if (! fp) {
    cout << "Error, file couldn't be opened" << endl; 
    return 1; 
}    
for(int i=0;i<80;i++)
{
    for(int j=0;j<434;j++)
    {
        if ( ! fp ) 
        {
            std::cout << "read error" << std::endl;
            break;
        }
        fp >> voice[i][j]
    }
}

auto options = torch::TensorOptions().dtype(at::kDouble);
auto tensor = torch::zeros({80,434}, options);
for (int i = 0; i < 80; i++)
    tensor.slice(0, i,i+1) = torch::from_blob(vect[i].data(), {434}, options);

现在张量是80 * 434，我怎样才能将这个张量中的一维添加到3d，我想要1 * 80 * 434

Answer 1

auto tensor = torch::zeros({80,434}, options);

在这一行之后，

auto tensor = tensor.view({1, 80, 434});

我建议在第二行中创建另一个变量而不是张量，例如：

auto transformed_tensor = tensor.view({1, 80, 434});