libtorch读取数组数据
相关函数:torch::from_blob
at::Tensor torch::from_blob(void *data, at::IntArrayRef sizes, at::IntArrayRef strides, const Deleter &deleter, const at::TensorOptions &options = at::TensorOptions())
Exposes the given data as a Tensor without taking ownership of the original data.
sizes should specify the shape of the tensor, strides the stride in each dimension. The deleter function (a std::function<void(void*)>) will be called on the data when the Tensor data would normally be deallocated. The TensorOptions specify additional configuration options for the returned tensor, such as what type to interpret the data as.
官网说明链接:torch::from_blob
1.读取浮点数组
函数默认以float32数据类型。
at::Tensor tensor = torch::from_blob(array, Height * Width * Chanels);
2.读取int32类型数组
at::Tensor tensor = torch::from_blob(array, Height * Width * Chanels, torch::TensorOptions().dtype(torch::kInt32)).clone();