Model forward pass

def forward(self, observations):
        """ Model forward pass """
        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