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）です。