TensorFlow
不浪不欢
创客,我用知识改变世界
展开
-
tf.cast()使用
tf.cast()函数是将tensorflow中数据类型转换为其他类型的函数:cast(x, dtype, name=None) y = 3 x = tf.cast(y, dtype=tf.float32) with tf.Session() as sess: print(sess.run(x))输出结果:3.0...原创 2019-07-03 10:07:24 · 404 阅读 · 0 评论 -
tf.argmax()和np.argmax()使用:
tf.argmax()和np.argmax()使用:tf.argmax是tensorflow用numpy的np.argmax实现的,用于返回矩阵或者向量中的最大元素所在的索引值,两者使用方法相同。import numpy as npx=np.array([[1,2,5],[2,1,1],[5,8,2]])y=np.argmax(x,0)#参数0表示列,参数1表示行print(x)pri...原创 2019-06-29 12:42:50 · 287 阅读 · 0 评论 -
tf.equal()用法
tf.equal(x, y)在矩阵或者向量x和y中,如果相同位置元素相等,返回True,否则返回False x=np.array([[1,2,3],[4,5,6]]) y=np.array([[1,5,3],[2,5,2]]) z=np.equal(x,y) print(z)运行结果如下[[ True False True] [False True ...原创 2019-06-29 12:55:03 · 1906 阅读 · 0 评论