Tensorflow Keras Neural Network With Cross