我正在尝试使用英特尔MKL计算一个3D数组的三维卷积。能否有人友好地给我一些提示，我该如何做到这一点？是否可以使用MKL来实现？提前致谢。
使用英特尔MKL进行3D卷积
3
- afp_2008
1
2如果您找到了解决方案，请发布出来。好问题。 - chappjc
1个回答
1
英特尔在他们的页面上有 一个三维FFT 的例子 , 这对于在频率空间中通过乘法执行卷积应该是有帮助的。 很抱歉我没有完整的解决方案:
三维实数FFT（C接口）
#include "mkl_dfti.h"
float x[32][100][19];
float _Complex y[32][100][10]; /* 10 = 19/2 + 1 */
DFTI_DESCRIPTOR_HANDLE my_desc_handle;
MKL_LONG status, l[3];
MKL_LONG strides_out[4];
//...put input data into x[j][k][s] 0<=j<=31, 0<=k<=99, 0<=s<=18
l[0] = 32; l[1] = 100; l[2] = 19;
strides_out[0] = 0; strides_out[1] = 1000;
strides_out[2] = 10; strides_out[3] = 1;
status = DftiCreateDescriptor( &my_desc_handle, DFTI_SINGLE,
DFTI_REAL, 3, l );
status = DftiSetValue(my_desc_handle,
DFTI_CONJUGATE_EVEN_STORAGE, DFTI_COMPLEX_COMPLEX);
status = DftiSetValue( my_desc_handle, DFTI_PLACEMENT, DFTI_NOT_INPLACE );
status = DftiSetValue(my_desc_handle,
DFTI_OUTPUT_STRIDES, strides_out);
status = DftiCommitDescriptor(my_desc_handle);
status = DftiComputeForward(my_desc_handle, x, y);
status = DftiFreeDescriptor(&my_desc_handle);
/* result is the complex value z(j,k,s) 0<=j<=31; 0<=k<=99, 0<=s<=9
and is stored in complex matrix y in CCE format. */
下一步将对填充后的内核执行相同的变换，两个复杂数组进行逐点乘法，然后进行反向FFT。
- chappjc
1
谢谢。我已经看过那个例子，它展示了如何使用该库。 - afp_2008
网页内容由stack overflow 提供, 点击上面的可以查看英文原文，
原文链接
原文链接