![](https://img-blog.csdnimg.cn/20201014180756928.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
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 · 417 阅读 · 0 评论 -
tf.argmax()和np.argmax()使用:
tf.argmax()和np.argmax()使用: tf.argmax是tensorflow用numpy的np.argmax实现的,用于返回矩阵或者向量中的最大元素所在的索引值,两者使用方法相同。 import numpy as np x=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 · 292 阅读 · 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 · 1930 阅读 · 0 评论