Предположим, у меня есть тензор в Tensorflow, его значения такие:
A = [[0.7, 0.2, 0.1],[0.1, 0.4, 0.5]]
Как я могу изменить этот тензор на следующее:
B = [[1, 0, 0],[0, 0, 1]]
Другими словами, я хочу просто сохранить максимум и заменить его на 1.
Буду признателен за любую помощь.