【tensorflow】tf-argmax()
import tensorflow as tfinput = [1, 3, 5, 7]output = ([[1, 2],[3, 5]])a = tf.argmax(input, 0) # 0表是按行b = tf.argmax(output, 1) # 1 表示按列with tf.Session() as sess:print("输出input最大值的索引下标：", sess.run(a))print("输出output最大值的索引下标,这是一个二维数组：",sess.run(b))
#rerurn:
3
[1 1]
本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！