code: def forward(self, observations):
          
          input_data = self.input_block(observations)
          advantage_features, value_features = self.backbone(input_data)
          log_histogram = self.q_head(advantage_features, value_features)
          return log_histogram
docs: Model forward pass