#tensorflow #pytorch
#tensorflow #pytorch
Вопрос:
overlap-> tensor([[0.0000, 0.0000, 0.0000, ..., 0.6466, 0.7945, 0.5389]],
device='cuda:0')
overlap_for_each_prior, object_for_each_prior = overlap.max(dim=0) # (8732)
this .max(dim= 0) возвращает два возврата.
есть ли какой-либо метод эквивалентности в tensorflow 2.0?
Комментарии:
1. Я думаю, это лучше спросить в Stack Overflow. Хотя обе эти библиотеки Python могут использоваться в контексте AI, вопрос не в AI, а в функции программных библиотек. В целом этот сайт посвящен математике и теории искусственного интеллекта, а не техническим особенностям инструментов.
Ответ №1:
Нет, вы должны использовать tf.math.argmax . Используя его, вы также можете получить максимальное количество элементов:
A = tf.constant([2, 20, 30, 3, 6])
maximum_index = tf.math.argmax(A)
A[maximum_index], maximum_index
# >>> 30, 2