class_docstr: An iterator to load sequence data.
class_docstr_tok: ["An","iterator","to","load","sequence","data","."]
code: def __init__(self, batch_size, num_steps, token_type):
      self.data_iter_fn = seq_data_iter
      self.corpus, self.vocab = load_shakespeare_corpus(token_type)
      self.batch_size, self.num_steps = batch_size, num_steps
  
imports: import collections
  import random
  import re
  import jax
  import jax.numpy
  import tensorflow_datasets
  import objax
  from objax.functional import one_hot
  from objax.zoo.rnn import RNN
label_desc: Use image node_id 1 to create a new DataLoader object  with example: obj = DataLoader(batch_size, num_steps, token_type)