我正在尝试在Tensorflow-Keras的自定义层中使用多个输入。使用方法可以是任何东西,现在它被定义为将掩码与图像相乘。我已经在Stack Overflow上搜索过,但是我找到的唯一答案是针对TF 1.x的,所以没有帮助。
class mul(layers.Layer):
def __init__(self, **kwargs):
super().__init__(**kwargs)
# I've added pass because this is the simplest form I can come up with.
pass
def call(self, inputs):
# magic happens here and multiplications occur
return(Z)
tf.keras.Layer.call()
的契约(https://www.tensorflow.org/api_docs/python/tf/keras/layers/Layer#call),该契约明确规定`inputs`应为多个输入张量的列表/元组。 - Grisha