TensorFlow 返回張量的最小值索引

tf.argmin

argmin (
     input ,
     axis = None ,
     name = None ,
     dimension = None
 )

定義在：tensorflow/python/ops/math_ops.py.

參考指南：數(shù)學(xué)>序列比較和索引

返回在張量的軸上的具有最小值的索引.

請注意,在關(guān)聯(lián)的情況下,返回值的身份不能保證.

ARGS：

• input：張量.必須是下列類型之一：float32,float64,int64,int32,uint8,uint16,int16,int8,complex64,complex128,qint8,quint8,qint32,half.
• axis：張量.必須是以下類型之一：int32,int64.當(dāng)類型為 int32 時,應(yīng)滿足：0 <= axis <rank(input).用來描述輸入張量的哪個軸減少.對于矢量,使用 axis = 0.
• name：操作的名稱(可選).

返回：

返回的張量類型為 int 64.