tf.argmax()函数中axis参数理解

tf.argmax是tensorflow用numpy的np.argmax实现的，它能给出某个tensor对象在某一维上的其数据最大值所在的索引值，常用于metric（如acc）的计算

tf.argmax()函数中有个axis参数（轴），该参数能指定按照哪个维度计算。
如 在矩阵的结构中，axis可被设置为0或1，分别表示
0：按列计算，1：行计算

下面是两个具体的示例，以便于直观理解（因tf.argmax()的axis的用法与numpy中的argmax相同，为方便起见，本文使用了numpy.argmax()用于示例）

axis=0

当axis=0时，以列为单位，第一列的最大值为索引为3的8，所以第一个值为3，以此类推


axis=1

当axis=1时，以行为单位，第一行的最大值为索引为2的3，所以第一个结果为2


最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成，浏览时请结合常识与多方信息审慎甄别。
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容