在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::data
或Couldn't find method at::Tensor::data
。我可以通过获取值auto value_array = tensor_name=accessor
,但是使用起来更容易*tensor_name[0].data
。您能告诉我如何使用张data<>()
量值int
吗?
我的类型也有同样的问题bool
。
谢谢,阿夫申
用于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; }