tf.math.add() with keras plot_model() 无法显示添加操作的两个输入
tf.math.add() with keras plot_model() cound not show the two inputs of the add operation
如果我在模型中添加一行，比如 p3 = tf.math.add(up_sample,x3_,name="up_sample + x3_")
在keras中plot_model，只显示第一个输入，如下图
[![在此处输入图片描述][1]][1]
我想知道如何使用 tensorflow2 显示 add() 函数的两个输入
[1]: https://i.stack.imgur.com/zjJ9o.png
尝试将 tf.math.add 包裹在 Lambda 层中：
import tensorflow as tf
input1 = tf.keras.layers.Input((2,))
input2 = tf.keras.layers.Input((2,))
output = tf.keras.layers.Lambda(lambda x: tf.math.add(x[0], x[1]), name='upsample')([input1, input2])
model = tf.keras.Model([input1, input2], output)
tf.keras.utils.plot_model(model, show_shapes=True)
如果我在模型中添加一行，比如 p3 = tf.math.add(up_sample,x3_,name="up_sample + x3_")
在keras中plot_model，只显示第一个输入，如下图
[![在此处输入图片描述][1]][1]
我想知道如何使用 tensorflow2 显示 add() 函数的两个输入 [1]: https://i.stack.imgur.com/zjJ9o.png
尝试将 tf.math.add 包裹在 Lambda 层中：
import tensorflow as tf
input1 = tf.keras.layers.Input((2,))
input2 = tf.keras.layers.Input((2,))
output = tf.keras.layers.Lambda(lambda x: tf.math.add(x[0], x[1]), name='upsample')([input1, input2])
model = tf.keras.Model([input1, input2], output)
tf.keras.utils.plot_model(model, show_shapes=True)