tensorflow:图(Graph)的核心数据结构与通用函数(Utility function)

1. 图(Graph)的核心数据结构

  • tf.Graph.__init__:建立一个空图;
  • tf.Graph.as_default():一个将某图设置为默认图,并返回一个上下文管理器,常与 with 结构相搭配;

    g = tf.Graph()
    with g.as_default():
      # Define operations and tensors in `g`.
      c = tf.constant(30.0)
      assert c.graph is g
    
    c = tf.constant(4.0)
    assert c.graph is tf.get_default_graph()
  • tf.Graph.finalized:返回True,如果图被完成

    • tf.Graph.finalize():完成图的构建,即将其设置为只读模式
    if not tf.Graph.finalized:
        tf.Graph.finalize()
  • tf.Graph.control_dependencies(control_inputs):定义一个控制依赖,并返回一个上下文管理器,也常与 with 结构搭配;

    with g.control_dependencies([a, b, c]):
        ....
        d = …
        e = …
    
    
    # `d` 和 `e` 将在 `a`, `b`, 和`c`执行完之后运行.
    
    
    def my_func(pred, tensor):
        with tf.control_dependencies([pred]):
            # 乘法操作(op)创建在该上下文,所以被加入依赖控制中
            #执行完pred之后再执行matmul
            return tf.matmul(tensor, tensor)
  • tf.Graph.device(device_name_or_function):定义运行图所使用的设备,并返回一个上下文管理器

    with g.device('/gpu:0'): ...
    with g.device('/cpu:0'): ...

转载于:https://www.cnblogs.com/mtcnn/p/9421620.html

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值