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