Python Numpy One caliente para las regiones

¿Cuál es la mejor manera de hacer esta 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]])

En otras palabras, ¿cómo decodificar One Hot array?