Python tf.compat.v1.arg_max用法及代碼示例

返回跨張量維度的最大值的索引。

用法

tf.compat.v1.arg_max(
    input, dimension, output_type=tf.dtypes.int64, name=None
)

參數

input 一個Tensor。必須是以下類型之一：float32 , float64 , int32 , uint8 , int16 , int8 , complex64 , int64 , qint8 , quint8 , qint32 , bfloat16 , uint16 , complex128 , half , uint32 , uint64 , bool。
dimension 一個Tensor。必須是以下類型之一：int32 , int64。 int32 或 int64，必須在 [-rank(input), rank(input)) 範圍內。說明要減少的輸入張量的維度。對於向量，使用維度 = 0。
output_type 一個可選的 tf.DType 來自：tf.int32, tf.int64。默認為 tf.int64 。
name 操作的名稱(可選)。

Tensor 類型為 output_type 。

請注意，在平局的情況下，不能保證返回值的身份。

用法：

import tensorflow as tf
a = [1, 10, 26.9, 2.8, 166.32, 62.3]
b = tf.math.argmax(input = a)
c = tf.keras.backend.eval(b)
# c = 4
# here a[4] = 166.32 which is the largest element of a across axis 0

相關用法

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