通过 Keras 缩小张量维数

Question

我发现我的模型有一个形状为(?,1,60)的张量。我想知道如何将其减少到(?,60)？不确定是否可以针对维度进行形状调整或展平。有什么帮助吗？

Answer 1

两层都可以工作，但在这种情况下，我更喜欢使用 keras.layers.Flatten。下面是一个示例：

from keras.layers import Input, Flatten
from keras.models import Model
import numpy as np
a = Input(shape=(1, 60))
b = Flatten()(a)
model = Model(inputs=a, outputs=b)
model.compile('sgd', 'mse')
pred = model.predict(x=np.ones(shape=(2, 1, 60)))
print(pred.shape)