Keras 中的非展平层

Question

我想创建一个简单的 Keras 神经网络，它接受一个维度的输入矩阵(rows, columns) = (n, m)，将矩阵展平到一个维度(n*m, 1)，通过多个任意层发送展平后的矩阵，并在最后一层，再次将矩阵取消展平到一个维度的(n, m)释放这最终矩阵作为输出之前。

我遇到的问题是我在keras.io页面上没有找到任何关于 Unflatten 层的文档，我想知道是否有这样一个看似标准的通用层不存在的原因。有没有更自然、更简单的方法来完成我的提议？

Answer 1

您可以将Reshape图层用于此目的。它接受所需的输出形状作为其参数，并将输入张量重塑为该形状。例如：

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)