3

我有一个std::vector<std::vector<double>>我想将它转换成torch::Tensorlibtorch 的地方。但是，似乎 ,torch::tensor()或torch::from_blob(), 不能用于此目的！

我尝试使用c10::ArrayRef然后使用它来将数据转换为 a 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];
   }
}
4

2 回答 2

3

最简单的方法是使用简单std::vector<double>的而不是向量的向量。您将拥有连续的记忆并且torch::from_blob可以工作（如另一个答案中所述）。

如果那不可能/不方便，我建议采用以下解决方法。我假设您的向量是一个(n,m)矩阵（即所有n向量具有相同的大小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);

编辑：您可能需要添加一个调用clone，以防您无法确保向量将超过张量（因为from_blob不拥有所有权，因此当向量被销毁时其数据将被删除）

于 2020-08-18T11:19:48.970 回答
2

我没有使用您提到的任何库，但是如果我猜的话，这些库可能期望一个连续的数组，而不是散布在堆周围的内存的小段。

所以转换std::vector<std::vector<double>>为std::vector<double>并将vec.data()指针传递给torch

std::vector<double> linearize(const std::vector<std::vector<double>>& vec_vec) {
    std::vector<double> vec;
    for (const auto& v : vec_vec) {
        for (auto d : v) {
            vec.push_back(d);
        }
    }
    return vec;
}
于 2020-08-18T11:17:04.033 回答