Python Numpy One quente para regiões

Qual é a melhor maneira de criar essa matriz codificada One Hot array([[[1, 0, 0], [1, 0, 0], [0, 1, 0]], [[0, 0, 1], [0, 1, 0], [1, 0, 0]]]) Como array([[0, 0, 1], [2, 1, 0]])Em outras palavras, como decodificar um array Hot?