cuComplex.h和exp（）

Question

Q0：

exp（）是否支持cuComplex.h？

Q1：

如何寫A = B * EXP（I * C），其中A，B，C是實數的相同的尺寸數組？這是正確的嗎？

主：

cuComplex A; 
float B; 
cuComplex c; 

內核：

c[idx] = (0, C[idx]); 

A[idx] = B[idx] * exp(c[idx]); 

Q2：

cuComplex包含2個浮點數這意味着我不得不分配2倍以上的存儲器然後原始矩陣。有什麼辦法可以創建純虛數？

Answer 1

cuComplex.h只提供cuComplex上的一些基本操作（主要是在CUBLAS和CUFFT庫中使用的那些操作），不支持指數函數。

您可以使用組件式算法自己實現指數。 cuComplex在x分量中存儲複數的實部，在y分量中存儲虛部。給定複數z = X + I * Y，指數可以被計算爲：

EXP（Z）= EXP（X）*（COS（Y）+ I * SIN（Y））

這導致以下CUDA代碼（未經測試）：

cuComplex my_complex_exp (cuComplex arg) 
{ 
    cuComplex res; 
    float s, c; 
    float e = expf(arg.x); 
    sincosf(arg.y, &s, &c); 
    res.x = c * e; 
    res.y = s * e; 
    return res; 
}