tf.argmax()表示返回最大值的索引号，axis为0 ，表示返回每列最大值索引号。axis为1 ，表示返回每行最大值索引号
结果为
tf.argmax()以及axis
原文：https://www.cnblogs.com/san333/p/10507402.html
