添加 Keras 层以按给定轴展平输入

Question

如果我的最后一个 Keras Layer X 输出一个看起来像的张量，例如像这样：

{
  {4, [22, 16,  11, 33], 24},
  {4, [12, 33,  87, 17], 14},
  {3, [92, 407, 25, 27], 34}
}

如何再添加一层 Y 层，将 X 的输出转换为：

{
  {4, 22, 16,  11, 33, 24},
  {4, 12, 33,  87, 17, 14},
  {3, 92, 407, 25, 27, 34}
}

Answer 1

其实有一种叫做 Flatten 的层可以满足你的需要（Keras 核心层文档here）：

from keras.layers import Flatten

# if you are using sequential model
...
model.add(Flatten())
...

# if you are using Model api
...
Y = Flatten()(X)
...