class_docstr: The Model for Atari env
  Args:
      obs_space (Box): observation space.
      act_space (Discrete): action space.
docstr: Get policy network prediction
  Args:
      obs (np.array): current observation
docstr_tok: ["Get","policy","network","prediction","Args",":","obs","(","np.array",")",":","current","observation"]
code: def policy(self, obs):
      """Get policy network prediction
      Args:
          obs (np.array): current observation
      """
      obs = obs / 255.0
      out = F.relu(self.conv1(obs))
      out = F.relu(self.conv2(out))
      out = F.relu(self.conv3(out))
  
      out = self.flatten(out)
      out = F.relu(self.fc(out))
      logits = self.fc_pi(out)
      return logits
  
file_path: PARL/examples/PPO/atari_model.py
imports: import parl
  import paddle.nn
  import paddle.nn.functional
label_desc: Use image node_id 3 for calling the AtariModel obj's underlying member method code with example usage: obj.policy(obs) and returns: logits