temsorflow使用笔记(自用)

tf.argmax(input, axis=None, name=None, dimension=None)

功能:查找最大值的索引

input:数据;

axis:定义按照某一条轴进行查找,如:

   axis=0 按x轴查找最大值的索引,axis=1 按y轴查找最大值的索引,以此类推2,3,4,...即为多维张量时

name:定义操作名称

注意:tf.argmax()与numpy.argmax()作用相同。

   argmin()正好与argmax作用相反,它用于查找最小值索引

 

posted @ 2019-09-27 20:23  坏人不会飞  阅读(269)  评论(0编辑  收藏  举报