python-3.x - 来自两个输入源的自定义 Tensorflow 层

Question

我正在尝试从应该执行此操作的两个“输入源”构建自定义 TensorFlow 层

exp(来源A)+ cos(来源B)

但是，我什至不知道如何设置编写这样的自定义层。

注:我真的很想学习/理解这是如何工作的，所以解决方法不是最佳的......

Answer 1

这是一种可能性

class custom_layer(tf.keras.layers.Layer):

    def __init__(self):
        super(custom_layer, self).__init__()
        pass

    def call(self, inputs):

        input1, input2 = inputs
        return tf.exp(input1) + tf.cos(input2)

inp1 = Input((10,))
inp2 = Input((10,))
x = custom_layer()([inp1,inp2])
x = Dense(1)(x)
model = Model([inp1,inp2],x)
model.compile('adam','mse')
model.summary()

X1 = np.random.uniform(0,1, (100,10))
X2 = np.random.uniform(0,1, (100,10))
y = np.random.uniform(0,1, 100)

model.fit([X1,X2],y, epochs=3)