For a vector, the softmax obtains the distribution over K classes.
the predicted distribution is y = softmax (Uv), where y is the predicted distribution, U ∈ R K×D is the classification matrix, and v is the vector representation of node.