def action(self, observations):
observations = self.input_block(observations)<EOL>policy_hidden = self.policy_backbone(observations)<EOL>action = self.action_head(policy_hidden)<EOL>return action<EOL>
Calculate value for given state