如何将Eigen张量输出为可被MATLAB读取的文本文件？

首先修正原代码中的几处错误：

• 变量名笔误：eXX.setZero();应改为X.setZero();，同理eYY→Y、eZZ→Z；
• 大小写错误：double LZ = 128;应改为double Lz = 128;，否则后续计算dz会出现未定义变量错误；
• 索引逻辑错误：Eigen张量的索引顺序为tensor(维度0索引, 维度1索引, 维度2索引)，原代码中X(k,i,j)的索引顺序与张量定义的(nx,ny,nz)维度不匹配，会导致元素存储混乱，需修正为对应维度的索引。

修正后的核心循环代码 #

// 初始化X/Y/Z张量
for(int i = 0; i< nx; i++){
    for(int j = 0; j< ny; j++){
        for(int k = 0; k< nz; k++){ 
            X(i,j,k) = i*dx;  // 对应nx维度的第i个元素
            Y(i,j,k) = j*dy;  // 对应ny维度的第j个元素
            Z(i,j,k) = k*dz;  // 对应nz维度的第k个元素
        }
    }       
}

// 计算Out张量
Eigen::Tensor<double, 3> Out(nx,ny,nz); 
Out.setZero(); 
for(int i = 0; i< nx; i++){
    for(int j = 0; j< ny; j++){
        for(int k = 0; k< nz; k++){ 
            Out(i,j,k) = sin(3. * a * Z(i,j,k)) * sin(a * X(i,j,k)) * cos(b * Y(i,j,k));
        }
    }       
}

自定义张量输出函数 #

Eigen默认的<<操作符会将3D张量展平为二维矩阵输出，导致MATLAB读取后无法直接识别为3D数组。因此需要自定义输出逻辑，按线性顺序输出所有元素，保证与MATLAB的reshape操作兼容：

void writeTensorToMatlabFile(const Eigen::Tensor<double, 3>& tensor, const std::string& filename) {
    std::ofstream file(filename);
    if (!file.is_open()) {
        std::cerr << "无法打开文件：" << filename << std::endl;
        return;
    }

    int nx = tensor.dimension(0);
    int ny = tensor.dimension(1);
    int nz = tensor.dimension(2);

    // 按维度顺序(i,j,k)输出所有元素，每行一个值
    for (int i = 0; i < nx; ++i) {
        for (int j = 0; j < ny; ++j) {
            for (int k = 0; k < nz; ++k) {
                file << tensor(i, j, k) << "\n";
            }
        }
    }

    file.close();
}

调用输出函数替换原输出逻辑 #

将原代码中的：

std::ofstream file("test.txt");
if (file.is_open())
{
    file << Out << '\n'; 
}

替换为：

writeTensorToMatlabFile(Out, "test.txt");

MATLAB读取代码 #

在MATLAB中读取文本文件后，通过reshape将一维数组转换为16×16×16的3D数组：

data = load('test.txt');
tensor = reshape(data, 16, 16, 16);

内容的提问来源于stack exchange，提问作者Mathew77