如何合並兩個 Keras 張量？ (?, 4) 和 (?, 4) 到 (?, 4, 2)

Question

在 Keras 模型中進行一些計算后，我得到了兩個張量。 它的形狀是(?, 4) 。 我想將它們合並成一個新的張量。 例如。 [1, 2, 3, 4]和[5, 6, 7, 8]到[[1, 5], [2, 6], [3, 7], [4, 8]] 。
我怎樣才能得到它？

Answer 1

您可以使用 tensorflow 后端的 stack 函數作為以下代碼：

import keras.backend as K
...
z = K.stack([x,y],axis=-1)

其中 x, y 的形狀是 (?,4); z 的形狀將是 (?,4,2)