当前位置:  开发笔记 > 运维 > 正文

火炬C ++:通过使用* .data <int>()获得整数张量的值

如何解决《火炬C++:通过使用*.data<int>()获得整数张量的值》经验,为你挑选了1个好方法。

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

我的类型也有同样的问题bool

谢谢,阿夫申



1> Fábio Perez..:

用于item()从张量中获取标量。

int main() {
  torch::Tensor tensor = torch::randint(20, {2, 3});
  std::cout << tensor << std::endl;
  int a = tensor[0][0].item();
  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 
#include                                      

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

    return 0;
}

推荐阅读
pan2502851807
这个屌丝很懒,什么也没留下!
DevBox开发工具箱 | 专业的在线开发工具网站    京公网安备 11010802040832号  |  京ICP备19059560号-6
Copyright © 1998 - 2020 DevBox.CN. All Rights Reserved devBox.cn 开发工具箱 版权所有