为 tensorflow reduce_mean 定义名称
tensorflow 440
原文标题 :Define name for tensorflow reduce_mean
我正在使用 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 中的此类操作层?
谢谢!