Python Keras中的去平坦层

我想创建一个简单的Keras神经网络，它接受维度

（行、列）=（n，m）

，将矩阵展平为维度

（n*m，1）

，通过多个任意层发送展平矩阵，在最后一层，再次将矩阵展平为维度

（n，m）

将此最终矩阵作为输出发布之前

我遇到的问题是，我没有在页面上找到任何关于非平台层的文档，我想知道这样一个看似标准的通用层不存在是否有原因。有没有一种更自然、更简单的方法来实现我的建议？

您可以使用图层来实现这一目的。它接受所需的输出形状作为参数，并将输入张量重塑为该形状。例如：

from keras.layers import Reshape

rsh_inp = Reshape((n*m, 1))(inp)  # if you don't want the last axis with dimension 1, you can also use Flatten layer

# rsh_inp goes through a number of arbitrary layers ...

# reshape back the output
out = Reshape((n,m))(out_rsh_inp)

---

如果你的答案解决了你的问题，请通过点击答案旁边的复选标记来接受它，把它标记为“答案”。