我正在使用 tensorflow 2.7.1，我正在嘗試為模型中的某個層定義一個名稱，如下所示：
tf.reduce_mean(some_other_tf, axis=1, name='my_name')
稍后編譯模型后，我想通過其名稱訪問該層，使用
model.get_layer("my_name")
但似乎沒有可用定義名稱的層，特定層的名稱類似于：
tf.math.reduce_mean_27
在 tensoflow 的早期版本中，可以使用以下方法選擇該層：
tf_op_layer_my_name
如何通過定義的名稱訪問 tensorflow 2.7.1 中的此類操作層？
謝謝！
uj5u.com熱心網友回復：
這是因為reduce_mean它不是一個實際的層，只是一個操作。您可以改為使用Lambda從函式定義層并為它們指定特定名稱。例如
mean_result = tf.keras.layers.Lambda(lambda x: tf.reduce_mean(x, axis=1),
name="some_layer_name")(layer_input)
現在您的模型應該有一個實際上名為“some_layer_name”而不是其他名稱的層。
轉載請註明出處，本文鏈接：https://www.uj5u.com/houduan/450727.html