tf

with tf.Session() as sess:

    with tf.device("/gpu:0"):

        .................

a是一个tensor,如果要输出a的值,则print(a.eval())

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

  • tensorflow是通过计算图的方式建立网络。比喻说明:结构:计算图建立的只是一个网络框架。编程时框架中不会出现...
    MachineLP阅读 5,527评论 0 1
  • 原文:www.cnblogs.com/welhzh/p/6607581.html tf.nn.conv2d是Ten...
    dopami阅读 3,707评论 0 0
  • 这篇文章是针对有tensorflow基础但是记不住复杂变量函数的读者,文章列举了从输入变量到前向传播,反向优化,数...
    horsetif阅读 4,913评论 0 1
  • 试问,在孩子的成长路上,我们可曾真的没有错过。30岁之后的日子都是快的,是悄然的。 我问娃:“爸爸打过你吗?”小丫...
    未闻典阅读 2,187评论 0 3
  • 姓名:罗婷婷 公司:海南蔚蓝时代实业有限公司 组别:第 420期 努力4组 (参加培训时的组名) 【日精进打卡第4...
    LT婷_420期阅读 1,229评论 0 0

友情链接更多精彩内容