argmax 如何用于三维数组？

Question

    x=tf.constant ([
      [[1,2,3],
       [4,5,6]],
       [[7,8,9],
       [10,11,12]],
       [[13,14,15],
       [16,17,18]]])
    #print(x)
    print(tf.math.argmax(x,axis=0))

结果：

tf.Tensor(
         [[2 2 2]
          [2 2 2]], shape=(2, 3), dtype=int64)

argmax() 如何用于 3D 阵列？ 请有人帮忙！！！

Answer 1

(0,0) 处的结果为您提供了具有最高数字的索引（沿轴 = 0）。
所以在这里你比较 1、7 和 13。因为 13 是最大的数字，所以结果在 (0,0) = 2
对于 (0,1) 处的结果，您比较 2、8 和 14 等等...