Modelling single neuron in TensorFlow