tensorflow создать модель

[dosyvany]

питон 3.8 tensorflow 2.3

подскажите как приблизительно собрать модель
# входные данные
Xt = [ # только одна 1 в елементе из 5 значений
[ [0,0,0,0,1], [0,0,1,0,0], [0,0,0,0,1], [1,0,0,0,0], [0,0,1,0,0] , [0,0,1,0,0] ], 1 ( 6 елементов)

...............................................................
[ [0,0,0,0,1],[0,0,0,0,1],[0,0,0,0,1],[0,0,0,0,1],[0,0,0,0,1] , [0,0,1,0,0] ] 500 ( 6 елементов)
]



Yt = [ # только одна 1 в елементе из 5 значений
[ [0,0,0,0,1], [0,0,1,0,0], [0,0,0,0,1] ], 1 ( 3 елемента)

...............................................................
[ [0,0,0,0,1],[0,0,0,0,1],[0,0,0,0,1] ] 500 ( 3 елемента)
]




# НЕ РАБОТАЕТ

act ='relu'
model = Sequential()
model.add(LSTM(6 , activation=act, batch_input_shape=(500, 6, 5),return_sequences=True))
model.add(LSTM(3, activation='softmax')) # предполагаю что ето последний слой


model.compile( optimizer='adam',loss='categorical_crossentropy',metrics=['acc'])



md = model.fit(Xt, Yt, epochs=500, batch_size=500, verbose=1, shuffle=False)





ValueError: Shapes (500, 6, 5) and (500, 3) are incompatible