c++ - 如何将std::vector >转换为Torch::Tensor？

Question

我有一个std::vector<std::vector<double>>，我想将其转换为libtorch中的torch::Tensor。但是，看来torch::tensor()或torch::from_blob()不能用于此目的!
我尝试使用c10::ArrayRef，然后通过执行torch::Tensor将其用于将数据转换为c10::ArrayRef<std::vector<std::vector<double>>> res(myvecs)，但这似乎也没有用，因为我似乎找不到找到将其转换为torch::Tensor的方法。
我应该如何在libtorch中进行此转换？除以下以外，我还有哪些其他选择:

auto tensor = torch::zeros({ 46,85 });
for (size_t i = 0; i < 46; i++)
{
   for (size_t j = 0; j < 85; j++)
   {
       tensor[i][j] = probs[i][j];
   }
}

Answer 1

最简单的方法是使用简单的std::vector<double>而不是 vector 的 vector 。您将拥有连续的内存，并且torch::from_blob可以工作(如其他答案中所述)。
如果这不可能/不方便，建议采取以下解决方法。我假设您的 vector 是(n,m)矩阵(即所有n vector 的大小都等于m):

int n = 5, m = 4;
// Just creating some dummy data for example
std::vector<std::vector<double>> vect(n, std::vector<double>(m, 0)); 
for (int i = 0; i < n; i++)
    for (int j = 0; j < m; j++)
        vect[i][j] = i+j;

// Copying into a tensor
auto options = torch::TensorOptions().dtype(at::kDouble);
auto tensor = torch::zeros({n,m}, options);
for (int i = 0; i < n; i++)
    tensor.slice(0, i,i+1) = torch::from_blob(vect[i].data(), {m}, options);

编辑:如果您不能确保 vector 不会超出张量，可能需要添加对 clone的调用(因为 from_blob没有所有权，因此当 vector 被销毁时，其数据将被删除)