如何合并两个Keras张量? (?,4)和(?,4)到(?,4,2)

时间:2018-03-14 01:35:34

标签: python merge keras tensor

在Keras模型中进行一些计算之后,我有两个张量。它的形状是(?,4)。 我想把它合并在一个新的张量中。 例如。 [1,2,3,4]和[5,6,7,8]到[[1,5],[2,6],[3,7],[4,8]] 我怎么能得到它?

1 个答案:

答案 0 :(得分:0)

您可以使用tensorflow后端的堆栈功能,如下代码:

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

x,y的形状是(?,4); z的形状为(?,4,2)