code: def set_states(self, states=None, value=None):
          
          assert self.binded and self.params_initialized
          self._curr_module.set_states(states, value)
docs: Sets value for states. Only one of states & values can be specified.
  
          Parameters
          ----------
          states : list of list of NDArrays
              Source states arrays formatted like ``[[state1_dev1, state1_dev2],
              [state2_dev1, state2_dev2]]``.
          value : number
              A single scalar value for all state arrays.