class_docstr: An immutable variant of the Python dict.
class_docstr_tok: ["An","immutable","variant","of","the","Python","dict","."]
docstr: Flattens this FrozenDict.
  
  Returns:
    A flattened version of this FrozenDict instance.
docstr_tok: ["Flattens","this","FrozenDict",".","Returns",":","A","flattened","version","of","this","FrozenDict","instance","."]
code: def tree_flatten_with_keys(self) -> Tuple[Tuple[Any, ...], Hashable]:
      """Flattens this FrozenDict.
  
      Returns:
        A flattened version of this FrozenDict instance.
      """
      sorted_keys = sorted(self._dict)
      return tuple(
          [
              (jax.tree_util.DictKey(k), self._dict[k])
              for k in sorted_keys
          ]
      ), tuple(sorted_keys)
  
imports: import collections
  from types import MappingProxyType
  from typing import Any, Dict, Hashable, Mapping, Tuple, TypeVar, Union
  import jax
  from flax import serialization
label_desc: Use image node_id 17 for calling the FrozenDict obj's underlying member method code with example usage: obj.tree_flatten_with_keys() and returns: tuple,  tuple