In the custom model, keras.layers.Embedding Output Shape is multiple

Question

I want to use keras.layers.Embedding in a customized sub-model. But output shape is \'multiple\'.Then I try to write a demo and test it

The results of the two methods