张量流中序列化权重的排序

Question

我有一个来自第三方的tensorflow冻结模型文件，其中初始层是具有以下属性的卷积层;

• 内核大小：(3, 3)
• 过滤器数量：32
• 输入渠道：3

因此，稍后的参数数量将为3 X 3 X 3 X32。现在，保存的文件确实将权重存储为形状为(3, 3, 3, 32)的4维数组。

但是，我不确定如何解释该数组。因此，该顺序代表通道x内核大小x内核大小x数量过滤器还是内核大小x内核大小x通道x数量过滤器。

之所以这样做，是因为我正在使用pytorch创建补充模型，并且需要读取保存的模型并将其加载到pytorch正向模型中。