Torch C++：使用 *.data<int>() 获取 int 张量的值

Question

在 C++ 版本的 Libtorch 中，我发现我可以通过 *tensor_name[0].data<float>() 获取浮点张量的值，其中我可以使用任何其他有效索引来代替 0。但是，当我通过在张量创建中添加选项at::kInt 来定义int 张量时，我无法使用此结构来获取张量的值，即*tensor_name[0].data<at::kInt>() 或*tensor_name[0].data<int>() 之类的东西不起作用，并且调试器一直说Couldn't find method at::Tensor::data<at::kInt> 或Couldn't find method at::Tensor::data<int>。 我可以通过auto value_array = tensor_name=accessor<int,1>() 获取值，但使用*tensor_name[0].data<int>() 更容易。你能告诉我如何使用data<>() 来获取int 张量的值吗？

bool 类型也有同样的问题。

Answer 1

使用 item<dtype>() 从张量中获取标量。

int main() {
  torch::Tensor tensor = torch::randint(20, {2, 3});
  std::cout << tensor << std::endl;
  int a = tensor[0][0].item<int>();
  std::cout << a << std::endl;
  return 0;
}

~/l/build ❯❯❯ ./example-app
  3  10   3
  2   5   8
[ Variable[CPUFloatType]{2,3} ]
3

以下代码打印 0（在 Linux 上使用稳定的 libtorch 测试）：

#include <torch/script.h>
#include <iostream>                                     

int main(int argc, const char* argv[])                  
{
    auto indx = torch::zeros({20},at::dtype(at::kLong));
    std::cout << indx[0].item<long>() << std::endl;

    return 0;
}