def head(self, phi):
    phi = phi.repeat(self.particles, 1, 1)
    value = self.fc_value(self.model_seed['value_z'], phi)
    advantage = self.fc_advantage(self.model_seed['advantage_z'], phi)
    q = (value.expand_as(advantage) + (advantage - advantage.mean((- 1), keepdim=True).expand_as(advantage)))
    return q
def head(self, phi):
    phi = phi.repeat(self.particles, 1, 1)
    value = self.fc_value(self.model_seed['value_z'], phi)
    advantage = self.fc_advantage(self.model_seed['advantage_z'], phi)
    q = (value.expand_as(advantage) + (advantage - advantage.mean((- 1), keepdim=True).expand_as(advantage)))
    return q