repacakage_hidden.py

def repackage_hidden(h):
    """Wraps hidden states in new Variables, to detach them from their history."""
    if type(h) == Variable:
        return Variable(h.data)
    elif type(h) == tuple:
        return tuple(repackage_hidden(v) for v in h)
    elif type(h) == list:
        return [ repackage_hidden(v) for v in h ]

def repackage_hidden_dnc(h):
  if h is None:
    return None

  if type(h) is list and h[0] is None:
    return [None]

  (chx, mhxs, _) = h[0]
  chx = repackage_hidden(chx)
  if type(mhxs) is list:
    mhxs = [dict([(k, repackage_hidden(v)) for k, v in mhx.items()]) for mhx in mhxs]
  else:
    mhxs = dict([(k, repackage_hidden(v)) for k, v in mhxs.items()])
  return [(chx, mhxs, None)]