cudnn - 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）。