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

如果我的最后一个Keras层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}
}

---

实际上，有一种称为“展平”的层适合您的需要（Keras核心层文档）：

---

你的意思是我可以将整个输出展平，然后使用重塑层将展平张量转换为（-1，6）？

from keras.layers import Flatten

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

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