Learning to Tokenize for Generative Retrieval