def sample(self, histogram_logits):
histogram_probs = histogram_logits.exp()  <EOL>atoms = self.support_atoms.view(<NUM_LIT:1>, <NUM_LIT:1>, self.atoms)  <EOL>return (histogram_probs * atoms).sum(dim=-<NUM_LIT:1>).argmax(dim=<NUM_LIT:1>)<EOL>
Sample from a greedy strategy with given q-value histogram