TensorFlow 返回张量的最小值索引
2018-08-31 14:33 更新
tf.argmin
argmin (
input ,
axis = None ,
name = None ,
dimension = None
)
定义在:tensorflow/python/ops/math_ops.py.
参考指南:数学>序列比较和索引
返回在张量的轴上的具有最小值的索引.
请注意,在关联的情况下,返回值的身份不能保证.
ARGS:
- input:张量.必须是下列类型之一:float32,float64,int64,int32,uint8,uint16,int16,int8,complex64,complex128,qint8,quint8,qint32,half.
- axis:张量.必须是以下类型之一:int32,int64.当类型为 int32 时,应满足:0 <= axis <rank(input).用来描述输入张量的哪个轴减少.对于矢量,使用 axis = 0.
- name:操作的名称(可选).
返回:
返回的张量类型为 int 64.