tf.argmax()以及axis解析
2018-01-10 10:37
176 查看
首先,明确一点,tf.argmax可以认为就是np.argmax。tensorflow使用numpy实现的这个API。
简单的说,tf.argmax就是返回最大的那个数值所在的下标。
这个很好理解,只是tf.argmax()的参数让人有些迷惑,比如,tf.argmax(array, 1)和tf.argmax(array, 0)有啥区别呢?
这里面就涉及到一个概念:axis。上面例子中的1和0就是axis。我先笼统的解释这个问题,设置axis的主要原因是方便我们进行多个维度的计算。
在实例面前,再多的语言都是苍白的呀,上例子!
比如:
test = np.array([[1, 2, 3], [2, 3, 4], [5, 4, 3], [8, 7, 2]]) np.argmax(test, 0) #输出:array([3, 3, 1] np.argmax(test, 1) #输出:array([2, 2, 0, 0]1
2
3
啥意思呢?
axis = 0:
你就这么想,0是最大的范围,所有的数组都要进行比较,只是比较的是这些数组相同位置上的数:
test[0] = array([1, 2, 3]) test[1] = array([2, 3, 4]) test[2] = array([5, 4, 3]) test[3] = array([8, 7, 2]) # output : [3, 3, 1]1
2
3
4
5
axis = 1:
等于1的时候,比较范围缩小了,只会比较每个数组内的数的大小,结果也会根据有几个数组,产生几个结果。
test[0] = array([1, 2, 3]) #2 test[1] = array([2, 3, 4]) #2 test[2] = array([5, 4, 3]) #0 test[3] = array([8, 7, 2]) #01
2
3
4
这是里面都是数组长度一致的情况,如果不一致,axis最大值为最小的数组长度-1,超过则报错。
当不一致的时候,axis=0的比较也就变成了每个数组的和的比较。
相关文章推荐
- tf.argmax()以及axis解析
- tf.argmax()以及axis解析
- tensorflow基础(2)-tf.argmax和axis
- tf.argmax(input, axis=None, name=None, dimension=None)
- DAGScheduler的原理剖析以及源码解析
- xml解析之dom解析以及两种方式的特点
- 张正友相机标定Opencv实现以及标定流程&&标定结果评价&&图像矫正流程解析(附标定程序和棋盘图)
- [原理][源代码解析]spring中@Transactional,Propagation.SUPPORTS,以及 Hibernate Session,以及jdbc Connection关系
- GitHub Desktop离线包以及使用解析
- rtmp协议简单解析以及用其发送h264的flv文件
- Android碎片事务提交transaction.commit()和transaction.commitnow()的区别以及源码完全解析
- Android 对Layout_weight属性完全解析以及使用ListView来实现表格
- auto_ptr源码解析以及例子 --auto_ptr_ref
- 解决VS2008 无法解析的外部符号 "int __cdecl _AfxInitManaged(void)“,以及无法打开预编译头Debug\XXX.pch
- 文字处理以及tf使用_android原生
- Jquery读取json文件和xml文件以及解析
- 微软BI 之SSIS 系列 - 在 SSIS 中使用 Web Service 以及 XML 解析
- 监听总结之解析问题二:由sqlnet.ora文件寻到EZCONNECT适配器程序来解析主机名为主机上的多个ip地址(*) 以及 tnsping是用来测试某个配置的监听程序是否启动
- 外卖库 xml 解析 以及坐标baidu 转换
- ViewDragHelper完全解析以及对自定义ViewGroup的作用