适配ArrayFire 3.6.4版本匹配滤波算法的维度不兼容问题

你遇到的这个问题，是因为ArrayFire在后续版本中收紧了隐式广播的规则——旧版本会自动把1×1的数组（标量形式的array）和高维数组运算时做隐式维度扩展，但3.6.4版本要求你显式指定这个广播操作，避免潜在的歧义。

针对你的代码，有几种简单的修复方式：

方法1：使用broadcastTo显式扩展维度 #

broadcastTo函数可以直接把源数组扩展到你指定的目标维度，正好匹配data.x_mat的尺寸：

array dx = broadcastTo(data.AntX(ii), data.x_mat.dims()) - data.x_mat;
array dy = broadcastTo(data.AntY(ii), data.x_mat.dims()) - data.y_mat;
array dz = broadcastTo(data.AntZ(ii), data.x_mat.dims()) - data.z_mat;

这个方法最直观，直接复用data.x_mat的维度信息，不需要手动计算尺寸，代码可读性也高。

方法2：使用tile函数平铺数组 #

tile会把输入数组重复指定的次数来扩展维度，你可以把单个元素的array平铺成Sx×Sy的尺寸：

array dx = tile(data.AntX(ii), data.x_mat.dims(0), data.x_mat.dims(1)) - data.x_mat;
array dy = tile(data.AntY(ii), data.x_mat.dims(0), data.x_mat.dims(1)) - data.y_mat;
array dz = tile(data.AntZ(ii), data.x_mat.dims(0), data.x_mat.dims(1)) - data.z_mat;

这里data.x_mat.dims(0)是Sx，dims(1)是Sy，和目标维度完全一致。

方法3：转换为原生标量后运算 #

因为data.AntX(ii)是单个值，你可以把它转换成C++原生的double类型，ArrayFire会自动把标量广播到数组的维度：

double antX_val = data.AntX(ii).scalar<double>();
double antY_val = data.AntY(ii).scalar<double>();
double antZ_val = data.AntZ(ii).scalar<double>();

array dx = antX_val - data.x_mat;
array dy = antY_val - data.y_mat;
array dz = antZ_val - data.z_mat;

这种方式代码最简洁，适合单个元素的场景，不需要额外的数组操作函数。

修改后的完整函数示例 #

这里用broadcastTo的方式修改你的核心代码：

void matched_filter_ArrayFire(SAR_data& data) {
    double c = 299792458.0; // speed of light
    // Determine the size of the phase history data
    int K = data.phdata.dims(0); // # of frequency bins per pulse
    int Np = data.phdata.dims(1); // # of pulses
    // Initialize the image with all zero values (complex)
    data.im_final = constant<cdouble>(0, data.x_mat.dims(), c64);
    array im_slices = constant<cdouble>(0, K, data.x_mat.dims(0), data.x_mat.dims(1), c64);
    array fspan = array(seq(0.0, K-1)) * data.deltaF;
    cdouble unit = {0, 1};

    for (int ii = 0; ii < Np; ii++) {
        // 显式广播天线位置到图像像素维度，解决尺寸不匹配问题
        array dx = broadcastTo(data.AntX(ii), data.x_mat.dims()) - data.x_mat;
        array dy = broadcastTo(data.AntY(ii), data.x_mat.dims()) - data.y_mat;
        array dz = broadcastTo(data.AntZ(ii), data.x_mat.dims()) - data.z_mat;

        array dR = sqrt(dx*dx + dy*dy + dz*dz) - data.R0(ii);
        // calculate the frequency of each sample in the pulse (Hz)
        array freq = data.minF(ii) + fspan;
        array tt = data.phdata(span,ii);

        // perform the Matched Filter operation
        gfor (array jj, K) {
            im_slices(jj,span,span) = tt(jj)*exp(unit*((double)(4.0*Pi/c)*freq(jj)*dR));
        }
        tt = sum(im_slices,0);
        data.im_final = data.im_final + moddims(tt, data.x_mat.dims());
    }
}

这三种方法都能解决你的维度不匹配问题，你可以根据自己的代码习惯选择其中一种。

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