Armadillo矩阵与LibTorch张量数据互转失败，求解决方法

问题根源 #

Armadillo默认采用**列优先（column-major）存储数据，而LibTorch张量默认是行优先（row-major）**存储，直接调用from_blob会因为存储顺序不匹配导致数据错乱，这就是你看到输出值异常的核心原因。

解决方案 #

1. Armadillo矩阵转LibTorch张量 #

方案1：指定列优先布局（推荐）

直接在创建张量时匹配Armadillo的存储顺序，无需修改数据：

arma::mat matrix{{1, 2, 3}, {7, 8, 9}};
cout << "armadillo matrix:\n";
cout << matrix << endl;

// 指定张量布局为列优先，与Armadillo存储逻辑对齐
auto t = torch::from_blob(matrix.memptr(), {2, 3}, torch::kColumnMajor);
cout << "libtorch tensor:\n";
cout << t << endl;

方案2：转置适配行优先需求

如果业务逻辑需要行优先的张量，可以先按列优先读取数据，再转置调整：

// 先按3行2列的列优先读取，再转置为2行3列的行优先张量
auto t = torch::from_blob(matrix.memptr(), {3, 2}, torch::kColumnMajor).t();

2. LibTorch张量转Armadillo矩阵 #

反向转换同样需要对齐存储顺序，示例代码：

// 创建行优先的LibTorch张量
torch::Tensor tensor = torch::tensor({{1,2,3}, {7,8,9}});
cout << "libtorch tensor:\n";
cout << tensor << endl;

// 先转置张量（将行优先转为列优先），再传入Armadillo
// 最后两个参数：false=不复制数据，true=让Armadillo接管内存（按需调整）
arma::mat matrix(tensor.t().data_ptr<double>(), tensor.size(1), tensor.size(0), false, true);
cout << "armadillo matrix:\n";
cout << matrix << endl;

关键注意事项 #

• 内存生命周期：使用from_blob时，Armadillo矩阵的生命周期必须长于张量，否则张量会访问已释放的内存；若需要独立张量，可调用t.clone()创建数据副本。
• 数据类型匹配：Armadillo的mat默认是double类型，你示例中LibTorch张量是FloatType，如需类型一致，可改用arma::fmat，或创建张量时指定类型：

  auto t = torch::from_blob(matrix.memptr(), {2, 3}, torch::kColumnMajor, torch::kDouble);
  

内容的提问来源于stack exchange，提问作者一碗给力嗯