在C++中定义的基本类型如何转换为Tensor

#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();

Guess you like

Origin blog.csdn.net/qq_38973721/article/details/121856408