Python tf.math.argmax用法及代碼示例

返回張量軸上具有最大值的索引。

用法

tf.math.argmax(
    input, axis=None, output_type=tf.dtypes.int64, name=None
)

參數

input 一個Tensor。
axis 一個整數，要減少的軸。默認為 0。
output_type 可選的輸出數據類型(tf.int32 或 tf.int64)。默認為 tf.int64 。
name 操作的可選名稱。

Tensor 類型為 output_type 。

在身份的情況下返回最小的索引。

例如：

A = tf.constant([2, 20, 30, 3, 6])
tf.math.argmax(A)  # A[2] is maximum in tensor A
<tf.Tensor:shape=(), dtype=int64, numpy=2>
B = tf.constant([[2, 20, 30, 3, 6], [3, 11, 16, 1, 8],
                 [14, 45, 23, 5, 27]])
tf.math.argmax(B, 0)
<tf.Tensor:shape=(5,), dtype=int64, numpy=array([2, 2, 0, 2, 2])>
tf.math.argmax(B, 1)
<tf.Tensor:shape=(3,), dtype=int64, numpy=array([2, 2, 1])>
C = tf.constant([0, 0, 0, 0])
tf.math.argmax(C) # Returns smallest index in case of ties
<tf.Tensor:shape=(), dtype=int64, numpy=0>

相關用法

注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.math.argmax。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。