Wie ist die Form/Reihenfolge der Convolutional Filter von Cludn?

Question

Cudnn speichert seine Filter in einem Array. Ich weiß, dass die Gesamtgröße des Arrays num_filters_out * num_filters_in * height_filter * width_filter sein muss. Aber wie ist das Array eigentlich bestellt?

Ist es (num_filters_out, num_filters_in, height_filter, width_filter) oder ist es eine andere Permutation? (Ich vermutete, dass man basierend auf der Signatur von cudnnSetFilter4dDescriptor)

Answer 1

Es ist normalerweise wie Sie sagten: "KCHW", d.h. (num_filters_out, num_filters_in, height_filter, width_filter).