如何在TensorFlow中比较两个字符串？含字符串张量元素对比场景

首先得提醒你：不能直接用Python的==运算符直接判断两个字符串张量是否相等后放到if语句里，因为TensorFlow的张量是计算图里的节点，直接==会返回一个布尔型张量，而不是Python原生的bool值，没法直接作为if的判断条件。下面给你两种符合需求的实现方式：

方法1：得到布尔张量（适合TensorFlow计算图内操作） #

如果你只是需要在TensorFlow的计算流程中使用比较结果，直接用tf.equal()函数就可以，它会返回一个布尔张量，表示对应位置的字符串是否相等：

import tensorflow as tf

# 注意：不要用str作为变量名，这是Python内置函数
text_str = tf.constant(['0001', '0013', '0021', '0001'], dtype=tf.string)
str_1 = text_str[0]
str_2 = text_str[1]

# 使用tf.equal比较字符串张量
equal_tensor = tf.equal(str_1, str_2)
print(equal_tensor.numpy())  # 输出：False

方法2：转换成Python布尔值（用于if/else判断） #

如果你确实需要把比较结果拿到Python逻辑中用if判断，那需要调用.numpy()方法把张量转换成Python原生的布尔值：

import tensorflow as tf

text_str = tf.constant(['0001', '0013', '0021', '0001'], dtype=tf.string)
str_1 = text_str[0]
str_2 = text_str[1]

# 先比较得到布尔张量，再转成Python bool
is_equal = tf.equal(str_1, str_2).numpy()
if is_equal:
    flag = True
else:
    flag = False

print(flag)  # 输出：False

另外补充一点：如果是要比较整个字符串张量中所有元素和某个值是否相等，tf.equal()也能直接处理，比如比较所有元素和'0001'：

all_equal = tf.equal(text_str, '0001')
print(all_equal.numpy())  # 输出：[ True False False  True]

内容的提问来源于stack exchange，提问作者Niraj Bhujel