5

我正在使用 Torch C++ 前端，并希望在其中有一个具有指定值的张量。要实现这一点，可以手动分配内存并设置值，然后使用torch::from_blob在内存块上构建张量，但对我来说似乎不够干净。

在本文档的最底部，我发现可以使用下标直接访问和修改数据。但是，这种方法的运行时间开销很大，可能是因为下标访问会将张量的元素视为 0-d 张量。以下代码在我的机器上将花费超过 2 秒（-O3优化级别），这对于现代 CPU 来说太长了。

    torch::Tensor tensor = torch::empty({1000, 1000});
    for(int i=0; i < 1000; i++)
    {
        for(int j=0 ; j < 1000; j++)
        {
            tensor[i][j] = calc_tensor_data(i,j);
        }
    }

有没有一种干净快速的方法来实现这个目标？

4

1 回答 1

8

在互联网上进行了数小时无果的搜索后，我脑海中浮现出一个假设，我决定试一试。事实证明，同一文档中提到的访问器也可以很好地作为左值，尽管文档中根本没有提到这个特性。下面的代码就好了，和直接操作原始指针一样快。

    torch::Tensor tensor = torch::empty({1000, 1000});
    auto accessor = tensor.accessor<float,2>();
    for(int i=0; i < 1000; i++)
    {
        for(int j=0 ; j < 1000; j++)
        {
            accessor[i][j] = calc_tensor_data(i,j);
        }
    }
于 2020-01-10T07:13:02.387 回答