我有一个形状为(?, ?, 128)的数组，我想将其转换为(?, ?, 128,1,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的维度。