Category "softmax"

how can I make a 2D softmax Layer?

I am making a network using keras library. Lets suppose that I have 2D matrix [ 0 0 1 2 0 1 2 5 1 0 0 1 ] what I want to do is obtaining the following matrix [