以argmax为例关于axis多维度的一点总结归纳

当初写的时候是用的TF，也是基于python的。所以希望在轴线维度上加深印象。

PS：在二维数组上而言.axis = 0即为竖; axis = 1为横向。记住即可，就不用下面这么麻烦了。

C = [[[16,6,3]，

      [4,6,5]

 

      [[8,7,9]，

      [11,12,10]]

]

对于三维数组：

使用tf.Session（）作为SESS：

    打印（sess.run（tf.argmax（C，0）））

    打印（sess.run（tf.argmax（C，1）））

打印（sess.run（tf.argmax（C，2）））

------------------------------------------

对轴= 0则取三维数组的第一维

即两个黄块数组进行对应比较。则16与8比较/ 6与7比较。得出相对较大者的索引

输出为：

[[0 1 1]

 [1 1 1]]

-------------------------------------------------- ---------------

对应轴= 1 .;则取三维数组的第二维：

即16与4比较/ 8与11比较：

输出为：

[[0 0 1]

 [1 1 1]]

-------------------------------------------------- --------------------------------

对应轴= 2即为取数组的第三维：16与6与3比较 - “0 4,6,5比较---”1

输出为：

[[0 1]

 [2 1]]

 

第一次写博客发布。如果错误请指正！