Question

我正在尝试对以下代码中名为dense3的两层使用相同的权重。我知道不允许使用dense3作为名称，因为图层需要具有唯一的名称。

最好的方法是在Keras中指定相同重量的不同层？

model = keras.Sequential([
    keras.layers.Flatten(input_shape=(28, 28)),
    keras.layers.Dense(128, activation=tf.nn.relu,name='dense_1'),
    keras.layers.Dense(10, activation=tf.nn.relu, name='dense_2'),
    keras.layers.Dense(10, activation=tf.nn.relu,name='dense_3' ),
    keras.layers.Dense(10, activation=tf.nn.relu,name='dense_3' ),
    keras.layers.Dense(10, activation=tf.nn.softmax)
])

Answer 1

sharedLayer = keras.layers.Dense(10, activation=tf.nn.relu,name='dense_3' )

model = keras.Sequential([
    keras.layers.Flatten(input_shape=(28, 28)),
    keras.layers.Dense(128, activation=tf.nn.relu,name='dense_1'),
    keras.layers.Dense(10, activation=tf.nn.relu, name='dense_2'),
    sharedLayer,
    sharedLayer,
    keras.layers.Dense(10, activation=tf.nn.softmax)
])

克拉斯的体重分享

1 个答案: