浮点数舍入问题

Question

我在 CUDA 中完成了论文，但我遇到了最后一个问题，这与舍入浮点数有关。

我有一个名为 bin 的整数变量，它是 x+y*X_dim 的编码。给定 bin，我想找到起源它的 x 和 y 坐标，这样我就可以进行对称性计算。这是我原来的程序：

float yaux,xaux;
    yaux=(float)floorf((float)bin/((float)DETECTOR_X_DIM));


    if(abs(yaux-floorf(yaux)) < 0.0001)
    yaux=floorf(yaux);
    else
    yaux=ceilf(yaux);


    xaux=(float)((float)(((float)bin/((float)DETECTOR_X_DIM))-(float)yaux)*((float)DETECTOR_X_DIM));

    return (int)xaux;

    if(abs(xaux-floorf(xaux)) < 0.0001)
    xaux=floorf(xaux);
    else
    xaux=ceilf(xaux);
    return (int)xaux;


    xaux = (float)DETECTOR_X_DIM - xaux -(float)1;

    return (int)xaux+(int)yaux*DETECTOR_X_DIM;

问题是它适用于检测器的某些箱，但不适用于其他箱（它返回 xaux 加 1）。有更好的方法吗？

先感谢您

Answer 1

我假设您的意思是 bin = x+y*X_dim 其中 x,y 是您希望在给定浮点数 bin 的情况下恢复的整数索引（不一定是整数）并且已知整数常量X_dim。我将此基于您的其他问题（CUDA 内核向量的长度基于 threadIdx 和 CUDA 3D 矩阵索引），其中你的代码更具可读性。在这种情况下，您需要 Matlab 的 sub2idx 函数 的简化等效项：

void Sub2Idx(int& x, int& y, float bin, int X_dim) {
y = (int)(bin / X_dim);
x = (int)(bin) - y*X_dim;
}

但是，我假设您想沿 x 轴向下舍入 bin 。双线性插值可能更合适，但对于这样一个不明确的问题来说，假设太多了。