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

通常就像你说的：“KCHW”，即（num_filters_out，num_filters_in，height_filter，width_filter）。