keras - 通过 Keras 减少张量维度

Question

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

标签： kerastensor

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)