2
Как получить индекс максимального элемента тензора вдоль выбранной оси в TensorFlow?Индекс максимального элемента в тензоре тензора TensorFlow
Как получить индекс максимального элемента тензора вдоль выбранной оси в TensorFlow?Индекс максимального элемента в тензоре тензора TensorFlow
Извините, уже разобрался. Это:
tf.argmax(input, dimension, name=None)