Python tf.case用法及代碼示例

創建案例操作。

用法

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

另見tf.switch_case。

pred_fn_pairs 參數是大小為 N 的對的列表。每對包含一個布爾標量張量和一個 python 可調用項，如果布爾值計算為 True，則創建要返回的張量。 default 是一個可調用的生成張量列表。 pred_fn_pairs 和 default(如果提供)中的所有可調用對象都應返回相同數量和類型的張量。

如果 exclusive==True ，則評估所有謂詞，如果多個謂詞評估為 True ，則會引發異常。如果 exclusive==False ，則執行在第一個計算結果為 True 的謂詞處停止，並立即返回相應函數生成的張量。如果沒有任何謂詞評估為 True，則此操作返回由 default 生成的張量。

tf.case 支持 tf.nest 中實現的嵌套結構。所有可調用對象必須返回相同的(可能是嵌套的)列表、元組和/或命名元組的值結構。單例列表和元組是唯一的異常：當由可調用對象返回時，它們被隱式解包為單個值。通過傳遞 strict=True 禁用此行為。

示例 1：

偽代碼：

if (x < y) return 17;
else return 23;

表達式：

f1 = lambda:tf.constant(17)
f2 = lambda:tf.constant(23)
r = tf.case([(tf.less(x, y), f1)], default=f2)

示例 2：

偽代碼：

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

表達式：

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)

v2 兼容性

pred_fn_pairs 可以是 v1 中的字典。但是，tf.Tensor 和 tf.Variable 在 v2 中不再是可散列的，因此不能用作字典的鍵。請改用列表或元組。