softmax_cross_entropy_with_logits keras