logits_name = self._get_logits_name()
return self.get_layer(x, logits_name)
def get_probs(self, x):
:param x: A symbolic representation of the network input.
After Change
:return: A symbolic representation of the logits
logits_name = self._get_logits_name()
logits_layer = self.get_layer(x, logits_name)
// Need to deal with the case where softmax is part of the
// logits layer
if logits_name == self._get_softmax_name():