Скажем, у меня есть партия бинарных тензоров размером [batch_size,7] типа float,
например
[[0, 1, 0, 0, 1, 0, 1],
[1, 0, 0, 1, 0, 0, 0]]
размера [2,7]
Я хочу преобразовать это в список горячих векторов из всех появляющихся 1. В нашем примере это будет
[[[0, 1, 0, 0, 0, 0, 0], [0, 0, 0, 0, 1, 0, 0], [0, 0, 0, 0, 0, 0, 1]],
[[1, 0, 0, 0, 0, 0, 0],[0, 0, 0, 1, 0, 0, 0]]]
В быстром методе