tensorflow在这里和numpy是一致的。这个dimension就是numpy里的axis。tf.argmax 参数可以表达成一个多维数组 a[][]...[],tf.argmax(a, i) 的结果是比 a 少一维的数组 b[ ]...[ ]...[ ]b的单个元素取值是这样的：b[][]...[][]...[]= max({ j | a[][]...[][j][]...[]})
