class Tensor {
public:
Tensor() = default;
void setTempData(void* data, std::size_t data_size) {
temp_data_.resize(data_size);
std::memcpy(temp_data_.data(), data, data_size);
}
const std::vector<char>& getTempData() {
return temp_data_;
}
private:
std::vector<char> temp_data_;
};
int main() {
std::vector<float> data(4);
data[0] = 1.1;
data[1] = 2.2;
data[2] = 3.3;
data[3] = 4.4;
Tensor a;
a.setTempData(data.data(), sizeof(float)*4);
const auto& b = a.getTempData();
std::vector<float> result(4);
std::memcpy(result.data(), b.data(), 16);
for (int64_t i = 0; i < 4; ++i) {
std::cout<< result[i] <<std::endl;
}
return 0;
}
c++数据保存方法
最新推荐文章于 2023-10-18 13:55:24 发布