如何解决Tensorflow2 - 如何为 tf_op_layer 分配名称?
我的模型结果之一是 tf_op_layer
,我需要为其分配一个结果名称。但是,我注意到参数名称通常不适用于 tf_op_layer
。
我面临的一个虚拟例子。在摘要中,可以看到 op_layer 的缺失名称。另一方面,它适用于密集层。
x_input = tf.keras.layers.Input(shape=(1,),name='input_test')
x = tf.keras.layers.Subtract(name='subtraction_1')([x_input,tf.constant([[12.]])])
x_out_1 = tf.keras.layers.Multiply(name='multiply_1')([x,tf.constant([[3.]])])
x_out_2 = tf.keras.layers.Dense(1,name='dense_1')(x)
model = tf.keras.models.Model(x_input,[x_out_1,x_out_2])
model.summary()
# Model: "model"
# __________________________________________________________________________________________________
# Layer (type) Output Shape Param # Connected to
# ==================================================================================================
# input_test (InputLayer) [(None,1)] 0
# __________________________________________________________________________________________________
# tf_op_layer_sub (TensorFlowOpLa [(None,1)] 0 input_test[0][0]
# __________________________________________________________________________________________________
# tf_op_layer_mul (TensorFlowOpLa [(None,1)] 0 tf_op_layer_sub[0][0]
# __________________________________________________________________________________________________
# dense_1 (Dense) (None,1) 2 tf_op_layer_sub[0][0]
# ==================================================================================================
# Total params: 2
# Trainable params: 2
# Non-trainable params: 0
# _________________________
我使用的是 TensorFlow 2.1.0 版。
还有其他方法可以为 op_layer 正确分配名称吗?
提前致谢,
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。