将tf.switch_case或tf.case与字符串条件一起使用

import tensorflow as tf

def case_1():
    return tf.constant("This is case 1")

def case_2():
    return tf.constant("This is case 2")

def case_3():
    return tf.constant("This is case 3")

def default_case():
    return tf.constant("This is the default case")

# 定义字符串条件和对应的操作
conditions = {
    "case1": case_1,
    "case2": case_2,
    "case3": case_3
}

# 定义字符串条件
input_condition = tf.constant("case2")

# 使用tf.switch_case根据字符串条件执行不同的操作
output = tf.switch_case(input_condition, conditions, default=default_case)

print(output.numpy())  # 输出：b'This is case 2'

import tensorflow as tf

def case_1():
    return tf.constant("This is case 1")

def case_2():
    return tf.constant("This is case 2")

def case_3():
    return tf.constant("This is case 3")

def default_case():
    return tf.constant("This is the default case")

# 定义条件和对应的操作
conditions = [
    (tf.equal(tf.constant(1), tf.constant(1)), case_1),
    (tf.equal(tf.constant(2), tf.constant(2)), case_2),
    (tf.equal(tf.constant(3), tf.constant(3)), case_3)
]

# 使用tf.case根据条件执行不同的操作
output = tf.case(conditions, default=default_case)

print(output.numpy())  # 输出：b'This is case 1'