code: def forward(self, observations):
          
          input_data = self.input_block(observations)
  
          base_output = self.backbone(input_data)
          policy_params = self.action_head(base_output)
  
          q = self.q_head(base_output)
  
          return policy_params, q
docs: Calculate model outputs