2020年11月4日
tensorflow GPU的使用
摘要： 参考：https://blog.csdn.net/mzpmzk/article/details/78647711 import os # 默认情况，TF 会占用所有 GPU 的所有内存, 我们可以指定 # 只有 GPU0 和 GPU1 这两块卡被看到，从而达到限制其使用所有GPU的目的 os.env 阅读全文
posted @ 2020-11-04 12:12 qiezi_online 阅读(179) 评论(0) 推荐(0)
理解 tf.reduce_sum()，以及tensorflow的维axis
摘要： 易错点：注意带上参数axis，否则的话，默认对全部元素求和，返回一个数值int 参考：https://www.jianshu.com/p/30b40b504bae tf.reduce_sum( input_tensor, axis=None, keepdims=None, name=None, re 阅读全文
posted @ 2020-11-04 10:07 qiezi_online 阅读(262) 评论(0) 推荐(0)