【tensorflow】 tf.argmax() 用法例子