libtorch如何获取张量设备类型？

Question

libtorch有没有什么方法来获取张量或模型设备类型。

就像pytorch里的xxx.device一样？

import torch
tensor = torch.rand(3,4)
print(tensor.device)

Answer 1

libtorch的设计目的是在C++中提供与python中几乎完全相同的功能，因此如果您有疑问，可以尝试：

#include <torch/torch.h>
torch::Tensor tensor = torch::rand({3,4});
std::cout << tensor.device() << std::endl;

曲线图:工作正常\o/