在 GPU 上处理复数

在 GPU 上处理复数的条件

如果在 GPU 上运行的函数的输出可能很复杂，则必须使用 complex 函数明确指定其输入参量为复杂。这适用于直接对 gpuArray 对象进行操作的函数，以及使用 arrayfun 对 gpuArray 数据进行操作的函数。

当创建含有负元素的 gpuArray 时，使用 G = gpuArray(complex(x))，则可成功执行 sqrt(G)。例如，

x = [-1 0 1];
G = gpuArray(complex(x));
rootG = sqrt(G)

   0.0000 + 1.0000i   0.0000 + 0.0000i   1.0000 + 0.0000i

如果结果是复数数据的 gpuArray，并且所有虚部都是零，则保留这些部分并且数据保持复数。这可能会导致在使用例如 sort 和 isreal 操作数据时执行不必要的计算。

如果使用 arrayfun，则可以在 arrayfun 调用中将函数输入指定为复杂。例如，

x = [-1 0 1];
G = gpuArray(x);
A = arrayfun(@(p) sqrt(complex(p)), G)

   0.0000 + 1.0000i   0.0000 + 0.0000i   1.0000 + 0.0000i

返回复杂数据的函数

下表列出了可能返回复数数据的函数，以及输出保持实数的输入范围。