def value(self, observations, state):
input_data = self.input_block(observations)<EOL>base_output, new_state = self.backbone(input_data, state)<EOL>value_output = self.value_head(base_output)<EOL>return value_output<EOL>
Calculate only value head for given state