def forward(self, observations, state):
input_data = self.input_block(observations)<EOL>base_output, new_state = self.backbone(input_data, state=state)<EOL>action_output = self.action_head(base_output)<EOL>value_output = self.value_head(base_output)<EOL>return action_output, value_output, new_state<EOL>
Calculate model outputs