python - keras 功能 API 中的 3D 输入到 2D 输出

Question

我有三个 Numpy 数组，每个数组的形状（6000,3,256）。我想将这三个用作密集神经网络的 Keras 功能 API 的输入。我的输出应该是形状（6000,2）。我对如何指定尺寸感到困惑。

标签： pythonnumpykerasdeep-learningneural-network

Answer 1

密集或全连接层采用一维输入，因此您需要展平输入数组。

X = np.reshape(6000, 3*256)

ip = Input(shape=(3*256))
dense1 = Dense(128, activation = 'relu')(ip)
dense2 = Dense(2, activation = 'sigmoid')(dense1)

您需要根据需要选择中间层、损失函数、激活。