Однократное представление матрицы в numpy

Question

Что является самым простым/умным способом перехода от матрицы значений к одному горячему представлению одной и той же вещи в 3d-тензоре? Например, если матрица является индексом после Argmax в тензором, как:

indices=numpy.argmax(mytensor,axis=2) 

Где тензор 3D [х, у, г] и индексы, естественно, будет [х, у]. Теперь вы хотите перейти к тензору 3D [x, y, z], который имеет 1s вместо maxes по оси = 2 и 0 в любом другом месте.

P.S. Я знаю ответ для вектора в 1-горячую матрицу, но это матрица для 1-горячего тензора.

Answer 1

Один из совершенных установок для использования broadcasting -

indices[...,None] == np.arange(mytensor.shape[-1]) 

Если вам нужно в Интс из 0s и 1s, добавьте с .astype(int)