如何在keras中将维度3的数组转换为5

Question

我有一个形状为(?, ?, 128)的数组，我想将其转换为(?, ?, 128,1,1)。

我尝试搜索它，但是没有发现任何好东西，或者也许我没有使用正确的词来找到自己想要的东西。 有什么有效的方法可以做到吗？

This是我发现的最相关的东西，与我想要的东西不一样。

提前感谢：）

Answer 1

使用Numpy，您可以reshape

a = np.random.rand(2, 3, 128)
b = a.reshape(a.shape + (1, 1))

使用Keras，您可以执行keras.backend.expand_dims(x, axis=-1)添加尺寸为1的尺寸。