(d_model, dropout, mode)
return tl.Parallel([], tl.Serial(
      ResidualSwitchUnit(d_model, dropout, mode),
      ReverseShuffleLayer(),
  ))
Takes (n_layer, state) and returns (n_layer, reverse_shuffle_layer(rsu(state))).
Takes (n_layer, state) and returns (n_layer, reverse_shuffle_layer(rsu(state))).
Takes ( n_layer state ) and returns ( n_layer reverse_shuffle_layer ( rsu ( state ))) .
def _BackwardStep(d_model, dropout, mode):
  """Takes (n_layer, state) and returns (n_layer, reverse_shuffle_layer(rsu(state)))."""
  return tl.Parallel([], tl.Serial(
      ResidualSwitchUnit(d_model, dropout, mode),
      ReverseShuffleLayer(),
  ))
def _BackwardStep ( d_model , dropout , mode ) : return tl . Parallel ( [ ] , tl . Serial ( ResidualSwitchUnit ( d_model , dropout , mode ) , ReverseShuffleLayer ( ) , ) )