#include <torch/extension.h>
int data [] = {
1, 2, 3, 4, 5};
auto op = torch::TensorOptions().dtype(torch::kInt8);
torch::Tensor data_torch = torch::from_blob(data, {
sizeof(data)/sizeof(int)}, op).clone();
在C++中定义的基本类型如何转换为Tensor
猜你喜欢
转载自blog.csdn.net/qq_38973721/article/details/121856408
今日推荐
周排行