首页
Java
登录
6mi
u
盘
搜
搜 索
Java
tensorflow(1):argmax的使用,并打印出tensor的值
tensorflow(1):argmax的使用,并打印出tensor的值
xiaoxiao
2021-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
)