TensorFlow第一天
配置一个gpu的TensorFlow真的是要了我的老命,一路上采坑无数,搞了一天终于完成了开始跑demo(真实菜鸡,太卑微了)…
写一点笔记吧…
1 .
print(‘before:’,a.dtype) # 读取原有张量的数值精度
if a.dtype != tf.float32: # 如果精度不符合要求,则进行转换
a = tf.cast(a,tf.float32) # tf.cast 函数可以...
原创
2020-01-04 10:53:02 ·
101 阅读 ·
0 评论