常用的函数: tf.argmax(input, axis=None, name=None, dimension=None) input:输入Tensor axis:0表示按列,1表示按行 name:名称 返回的为索引 import tensorflow as tf import numpy as np A , B = [[1, 3, 4, 5, 6]],[[1, 3, 4], [2, 4, 1]] with tf.Session() as sess: print(sess.run(tf.argm