TensorFlow 返回张量维度上最大值的索引

tf.arg_max


arg_max (
input ,
dimension ,
name = None
)

返回在张量维度上具有最大值的索引.

请注意,在关联的情况下,返回值的身份不能保证.

ARGS:

  • input:张量.必须是下列类型之一:float32,float64,int64,int32,uint8,uint16,int16,int8,complex64,complex128,qint8,quint8,qint32,half.
  • dimension:张量.必须是以下类型之一:int32,int64.当类型为 int32 时,应满足:0 <= dimension <rank(input).描述输入张量的哪个维度可以减少.对于向量,请使用 dimension = 0.
  • name:操作的名称(可选).

返回:

返回的张量类型为 int64.

以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号

意见反馈
返回顶部