:return: A symbolic representation of the output logits (i.e., the
values fed as inputs to the softmax layer).
return self.get_layer(x, "logits")
def get_probs(self, x):
:param x: A symbolic representation of the network input
After Change
:return: A symbolic representation of the output logits (i.e., the
values fed as inputs to the softmax layer).
return self.fprop(x, **kwargs)[self.O_LOGITS]
def get_probs(self, x, **kwargs):
:param x: A symbolic representation of the network input