【发布时间】:2018-03-14 01:35:34
【问题描述】:
在 Keras 模型中进行一些计算后,我得到了两个张量。它的形状是(?, 4)。
我想把它们合并成一个新的张量。
例如。 [1, 2, 3, 4] 和 [5, 6, 7, 8] 到 [[1, 5], [2, 6], [3, 7], [4, 8]]。
我怎样才能得到它?
【问题讨论】:
-
试试
stack....
在 Keras 模型中进行一些计算后,我得到了两个张量。它的形状是(?, 4)。
我想把它们合并成一个新的张量。
例如。 [1, 2, 3, 4] 和 [5, 6, 7, 8] 到 [[1, 5], [2, 6], [3, 7], [4, 8]]。
我怎样才能得到它?
【问题讨论】:
stack ....
您可以使用 tensorflow 后端的 stack 函数,代码如下:
import keras.backend as K
...
z = K.stack([x,y],axis=-1)
其中x,y的形状是(?,4); z 的形状将是 (?,4,2)
【讨论】: