Уменьшение тензорного измерения с помощью Кераса

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)