c++ libtorch Eigen矩阵转换为torch张量时数据类型必须一致

最新推荐文章于 2025-11-22 23:45:52 发布

原创最新推荐文章于 2025-11-22 23:45:52 发布 · 302 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#c++

torch 同时被 2 个专栏收录

21 篇文章

订阅专栏

Eigen

10 篇文章

订阅专栏

错误代码

void torch_test8() {

    Eigen::MatrixXd a(Eigen::MatrixXd::Random(3, 3));
    std::cout << "a" << std::endl << a << std::endl;
    auto options = torch::TensorOptions().dtype(torch::kFloat32).device(torch::kCPU);
    torch::Tensor x = torch::from_blob(a.data(), { a.cols(), a.rows() }, options);
    std::cout << "x" << std::endl << x << std::endl;


}

输出结果

正确代码

void torch_test8() {

    Eigen::MatrixXd a(Eigen::MatrixXd::Random(3, 3));
    std::cout << "a" << std::endl << a << std::endl;
    Eigen::MatrixXf f = a.cast<float>(); 
    auto options = torch::TensorOptions().dtype(torch::kFloat32).device(torch::kCPU);
    torch::Tensor x = torch::from_blob(f.data(), { f.cols(), f.rows() }, options);
    std::cout << "x" << std::endl << x << std::endl;


}

Eigen的double矩阵不能从内存拷贝成float张量，必须先转换为float矩阵，再从内存拷贝为float张量