在具有以下部分架构的模型中:
我们在这里说Dense1和Dense2层(完全连接)。
我需要获得已经训练好的Dense2模型的权重,但是我在这一层得到的形状是(128,2048),这使我感到困惑,因为该层的理论解释是它应该为我的数组提供一个单一值(此处:2048)。我将Dense2的权重打印为:
weights = model.layers[layer_idx].get_weights()
我不应该看到建筑的形状吗? (无,2048)
最佳答案
这里,在第二个密集层中有2048个神经元,每个神经元都与上一层输出相连。因此,对于单个神经元,将有128个前一层神经元起作用。y = w1*x1 + w2*x2 + .......... + w128*x128
因此,总共(128 * 2048)= 262144个参数,其中2048个偏置向量总共为264192(262144 + 2048)个参数。
谢谢
关于python - 致密层重量形状,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/50212330/