Modification to return logit in inference_small()

[takerum]

I found the missing line for returning logit in inference_small(). I added the line at the end of inference_small().

[JulesGM]

This should be accepted. I also just stumbled into this, and came to the same conclusion:

inference_small doesn't return anything, yet in the cifar example, it's (non-existant) value is assigned to a variable. I'm pretty sure it's just that "return" is missing on the last line of inference_small, before inference_small_config(x, c).