docstr: Generate GRU's initial hidden states for prediction (batch_size=1)
          
docstr_tok: ["Generate","GRU","'s","initial","hidden","states","for","prediction","(","batch_size=1",")"]
code: def reset_agent(self):
      """Generate GRU's initial hidden states for prediction (batch_size=1)"""
      self.last_hidden_states = np.zeros(
          (self.n_agents, self.rnn_hidden_dim), dtype="float32"
      )
  
file_path: PARL/benchmark/fluid/QMIX/qmix_agent.py
imports: import parl
  from parl import layers
  import paddle.fluid
  import numpy
  from utils import AvailableActionsSampler
  from parl.utils import machine_info
label_desc: Use image node_id 2 for calling the QMixAgent obj's underlying member method code with example usage: obj.reset_agent() without return types