tensorflow(1):argmax的使用,并打印出tensor的值

xiaoxiao2021-02-28  72

tensorflow的argmax和numpy的argmax一样:

import tensorflow as tf import numpy as np a = np.array([[1,2,3],[1,2,3],[9,8,7]]) mm = tf.argmax(a,1) sess = tf.Session() print sess.run(mm) bb = np.argmax(a,1) print bb 输出:

[2 2 0] [2 2 0]

转载请注明原文地址: https://www.6miu.com/read-72109.html

最新回复(0)