如何在tensorflow2的keras密集層中逐元素地減去？

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)

或者你必須創建一個自定義層。