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