0

libtorch 是否有任何方法可以获取张量或模型设备类型。

就像xxx.device在pytorch中一样?

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

1 回答 1

2

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

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

情节扭曲:它有效\o/

于 2020-07-29T13:54:34.910 回答