要从C/C++数组中创建torch::Tensor,可以使用torch::from_blob()函数。以下是一个示例代码:
#include
int main() {
// 定义一个C数组
float data[] = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0};
// 获取C数组的大小
int64_t size = sizeof(data) / sizeof(float);
// 创建torch::Tensor对象,指向C数组的数据
torch::Tensor tensor = torch::from_blob(data, {size});
// 打印tensor
std::cout << "Tensor: " << tensor << std::endl;
return 0;
}
在上面的示例中,我们首先定义了一个C数组data
。然后,我们计算了数组的大小,并使用torch::from_blob()
函数创建了一个torch::Tensor对象tensor
。最后,我们打印了该tensor。