Сделаю?

ли способ найти верхk значения в двумерном тензоре в Tensorflow?

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

Большое спасибо.