cudnn卷积滤波器的形状/顺序是什么？

Question

Cudnn将其过滤器存储在一个数组中。我知道数组的总大小必须是num_filters_out * num_filters_in * height_filter * width_filter。但实际上阵列是如何排序的呢？

是(num_filters_out, num_filters_in, height_filter, width_filter)，还是不同的排列？ （我猜这个基于cudnnSetFilter4dDescriptor）

的签名

Answer 1

通常如你所说：＆＃34; KCHW＆＃34;，即（num_filters_out，num_filters_in，height_filter，width_filter）。