Разделение веса в Керасе

Я пытаюсь использовать один и тот же вес для двух слоев, которые в нижеследующем коде названы плотность3. Я знаю, что использование ' density3 ' в качестве имени не допускается, поскольку слои должны иметь уникальные имена. Что лучше всего было указать одинаковый вес разных слоев в керасе

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)
])

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)
])
Другие вопросы по тегам