code: def double_flatten_states(states):
      # XXX: Can probably write this in a nicer way
      states = flatten_states(
          [flatten_states(inner) for inner in states]
      )
      return [
          hidden.view([-1] + list(hidden.shape[2:]))
          for hidden in states
      ]
  
imports: import numbers
  import warnings
  from collections import namedtuple
  from typing import List, Tuple
  import torch
  import torch.jit
  import torch.nn
  from torch import Tensor
  from torch.nn import Parameter
label_desc: Use image node_id 6 for calling a global function with example usage: double_flatten_states(states) and returns: unknown