Stateless RNN vs stateful RNN - what should be the right batch_input_shape?

Question

stateless RNN model

model = tf.keras.models.Sequential([
         tf.keras.layers.Embedding(input_dim=(vocab_size), output_dim=256,mask_zero=True),
         tf.