TensorFlow 2.0
TF 2.0引入了一个称为AutoGraph的功能 , 该功能使您可以通过JIT将python代码编译为Graph执行。这意味着您可以使用python控制流语句(是的,这包括if
语句)。从文档中
签名支持常用的Python之类的语句
while
,for
,if
,break
,continue
和return
,以嵌套的支持。这意味着您可以在while
和if
语句的条件下使用Tensor表达式,或者在for
循环中遍历Tensor。
您将需要定义一个实现逻辑的函数,并使用tf.function
对其进行tf.function
。这是文档中的修改示例:
import tensorflow as tf
@tf.function
def sum_even(items):
s = 0
for c in items:
if tf.equal(c % 2, 0):
s += c
return s
sum_even(tf.constant([10, 12, 15, 20]))
# <tf.Tensor: id=1146, shape=(), dtype=int32, numpy=42>
0
假设我有以下代码:
if
语句在计算中if
工作(我认为不行)?如果没有,如何将if
语句添加到TensorFlow计算图中?