I figured all indices have the same max_index so that it can be converted to one-hot vectors, which makes most sense to me at least from an NLP perspective