def init_fromvector(vec, state): "" vals = [] var_sizes = state.size() i = 0 for var in state: vals.append(np.squeeze(vec[i:(i + var_sizes[var])])) i += var_sizes[var] return State(zip(state.keys(), vals))
def init_fromvector(vec, state): "" vals = [] var_sizes = state.size() i = 0 for var in state: vals.append(np.squeeze(vec[i:(i+var_sizes[var])])) i += var_sizes[var] return State(zip(state.keys(), vals))
def fromvector(self, vec): """ Update the state using a numpy array. :param vec: np.array for updating the state. """ var_sizes = self.size() i = 0 for var in self: self[var] = np.squeeze(vec[i:(i + var_sizes[var])]) i += var_sizes[var] return self
def fromvector(self, vec): """ Update the state using a numpy array. :param vec: np.array for updating the state. """ var_sizes = self.size() i = 0 for var in self: self[var] = np.squeeze(vec[i:(i+var_sizes[var])]) i += var_sizes[var] return self