code: def value(self, observations):
          
          input_data = self.input_block(observations)
          base_output = self.value_backbone(input_data)
          value_output = self.value_head(base_output)
          return value_output
docs: Calculate only value head for given state