zoukankan      html  css  js  c++  java
  • tf,cond()函数

      tensorflow里面有太多的函数功能,完全像是新学一门语言,所以打算采用,遇到不会的函数就记录下来,作为自己的函数手册。这篇要讲的就是tf,cond()。

    tf,cond(A,B,C) 就像是条件语句:if A: B else:C,举个例子:

    z = tf.multiply(a, b)
    result = tf.cond(x < y, lambda: tf.add(x, z), lambda: tf.square(y))
    

      解读一下这个官方的例子,首先是给z赋值了一个简单的乘法函数,然后调用tf,cond这个函数,它的意思是,如果 x<y,就执行函数 tf.add(x,z) , 否则 就执行 tf.square(y)。 在这里有liangdian注意:

      第一点,条件判断后执行命令都采用匿名函数的方式;

      第二点,因为tensorflow是计算图的形式,所以其实无论x与y的大小如何,后面的两条命令都是运行过的,只是最后根据x,y的大小决定数据流图走哪条路。

      tf,cond()函数,一般就掌握到能看懂代码,知道这个代码可以简化代码就够了,具体细节知道就好了,忘了也没事。下面再列两个我遇到的例子:

    # remove alpha channel if present
    input_image = tf.cond(tf.equal(tf.shape(input_image)[2], 4), lambda: input_image[:,:,:3], lambda: input_image)
    # convert grayscale to RGB
    input_image = tf.cond(tf.equal(tf.shape(input_image)[2], 1), lambda: tf.image.grayscale_to_rgb(input_image), lambda: input_image)
    

     

  • 相关阅读:
    IDEA使用总结1-Github下载代码和上传代码到Git
    Mysql-安装指南
    Dubbo安装及其实战1
    分布式设计(学习内容目录--后端架构师)
    elasticsearch安装指导(new)
    浅谈TCP/IP(new 常见面试问题)
    浅谈常用的设计模式(new)
    浅谈Elasicsearch
    浅谈数据库分库分表(new)
    JAVA 电子书下载地址
  • 原文地址:https://www.cnblogs.com/happy-sir/p/11530881.html
Copyright © 2011-2022 走看看