在tensorflow2.0中有pytorch的max操作吗？

问题描述

overlap-> tensor([[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)

此.max（dim = 0）返回两个返回值。
tensorflow 2.0中有什么等效方法吗？

解决方法

否，您必须使用tf.math.argmax。使用它，您还可以获得最大元素：

A = tf.constant([2,20,30,3,6])
maximum_index = tf.math.argmax(A)
A[maximum_index],maximum_index
# >>> 30,2