如何在tensorflow2的keras Dense层中逐元素减去？

Question

假设我在Dense(2) 层。如何添加层以首先记录整个张量，然后从第一列中逐元素减去第二列？谢谢。

array([[1,2],
       [3,4],
       [5,6]])

becomes

array([[log(2)-log(1)],
       [log(4)-log(3)],
       [log(6)-log(5)]])

Answer 1

我会这样做：

input = tf.keras.layers.Input(shape=(2,), dtype=tf.float32)
x = tf.keras.layers.Dense(2)(input)
x = tf.math.log(x[-1][0]) - tf.math.log(x[-1][1])
model = tf.keras.Model(inputs=input, outputs=x)

或者你必须创建一个自定义层。