Customize Softmax activation function in tensorflow

Question

I built a neural network in tensorflow which should predict the probabilities of an event to happen. As output function I therefore passed softmax. I tried to customize the soft