class_docstr: Random number generator module.
class_docstr_tok: ["Random","number","generator","module","."]
docstr: The random generator state (a tensor of 2 int32).
docstr_tok: ["The","random","generator","state","(","a","tensor","of","2","int32",")","."]
code: def key(self):
      """The random generator state (a tensor of 2 int32)."""
      if self._key is None:
          self._key = RandomState(self.initial_seed)
      return self._key
  
file_path: objax/objax/random/random.py
imports: from typing import Optional, Tuple
  import jax.random
  from objax.module import Module
  from objax.util import class_name
  from objax.variable import RandomState, VarCollection
label_desc: Use image node_id 2 for calling the Generator obj's underlying member method code with example usage: obj.key() and returns: self