tensorflow條件語句-tf.case

tf.case
tf.case(
    pred_fn_pairs,
    default=None,
    exclusive=False,
    strict=False,
    name='case'
)

創建case操作

pred_fn_pairs參數是大小爲N的字典或pairs的列表。每對包含一個布爾標量tensor和一個python可調用函數項，當條件爲True將返回對應的函數項創建的tensors。在pred_fn_pairs對中的所有調用子以及默認值（如果提供的話）都應該返回相同數量和類型的張量。

如果exclusive==True，則計算所有的謂詞，如果多個謂詞計算爲True，則引發異常。如果exclusive==False，則執行在求值爲True的第一個謂詞處停止，並且立即返回由相應函數生成的張量（tensors）。如果沒有一個謂詞評估爲true，則此操作返回默認生成的張量。

tf.case支持在tensorflow.python.util.nest中實現的嵌套結構。所有的調用都必須返回相同的（可能嵌套的）列表、元組和/或命名元組的值結構。單例列表和元組形成了唯一的例外：當由可調用程序返回時，它們被隱式地解包爲單個值。通過通過傳遞strict=True禁用此行爲。

如果無序字典在pred_fn_pairs使用，則條件測試的順序不能保證。不管怎麼樣，順序保證是確定的，以便在條件分支中變量按固定順序被創建。

#!/usr/bin/env python2
# -*- coding: utf-8 -*-
"""
Created on Mon Aug 27 11:16:32 2018
@author: myhaspl
"""

import tensorflow as tf
x = tf.constant(7)
y =  tf.constant(27)
f1 = lambda: tf.constant(17)
f2 = lambda: tf.constant(23)
r = tf.case([(tf.less(x, y), f1)], default=f2)
#if (x < y) return 17;
#else return 23;
sess=tf.Session()
with sess:

    print sess.run(r) 

17

Example 2:

#!/usr/bin/env python2
# -*- coding: utf-8 -*-
"""
Created on Mon Aug 27 11:16:32 2018
@author: myhaspl
"""

import tensorflow as tf
x = tf.constant(7)
y =  tf.constant(27)
z = tf.constant(21)
def f1(): 
    return tf.constant(17)
def f2(): 
    return tf.constant(23)
def f3(): 
    return tf.constant(-1)
r = tf.case({tf.less(x, y): f1, tf.greater(x, z): f2},default=f3, exclusive=True);

sess=tf.Session()
with sess:

    print sess.run(r)

相當於：

if (x < y && x > z) raise OpError("Only one predicate may evaluate true");
if (x < y) return 17;
else if (x > z) return 23;
else return -1;

參數:

pred_fn_pairs: 字典或pairs的列表（由boolean標量和可調用函數返回張量列表）
default: 默認返回tensors列表
exclusive: 如果爲True表示僅最多一個謂語爲True 
strict: boolean打開或關閉'strict' 模式
name: 操作的名字（可選）
返回:

第一個謂詞爲True時執行返回的tensors，如果沒有謂詞爲True，則返回default

Raises:

TypeError: pred_fn_pairs 不是一個列表或字典
TypeError: pred_fn_pairs是一個列表，但不包括2個元素的元組 
TypeError: fns[i] 不是任何i的調用，或default不是可調用的