Сделаю?
ли способ найти верхk
значения в двумерном тензоре в Tensorflow?
я могу использоватьtf.nn.top_k
для 1-D тензора, но он не может работать с 2-D тензором. У меня есть двумерный тензор с неизвестным размером, есть ли способ найти верхk
ценности и их показатели?
Большое спасибо.