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?