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 (将#修改为@）