ClassDecoder[probability]

decodes a probability vector of 1 and 0 into Integers of 0 to the value of the last dimension of probability minus one.

ClassDecoder[probability,nClass]

decodes a probability vector of 1 and 0 into Integers of 0 to nClass - 1.