def forward(self, observations):
input_data = self.input_block(observations)<EOL>base_output = self.backbone(input_data)<EOL>log_histogram = self.q_head(base_output)<EOL>return log_histogram<EOL>
Model forward pass