如何在Tensorflow中比较两个字符串？

Question

假设我想与字符串类型张量的元素进行比较：文本

str = tf.constant(['0001', '0013', '0021', '0001'], dtype=tf.string)
str_1 = str[0]
str_2 = str[1]
if str_1==str_2:
    flag=True
else:
    flag=False

Answer 1

您可以使用tf.math.equal()比较两个字符串张量。

代码：

tf.enable_eager_execution()

str_var = tf.constant(['0001', '0010', '0020', '0001'], dtype=tf.string)
str_1 = str_var[0]
str_2 = str_var[1]
str_3 = str_var[3]

print(tf.math.equal(str_1, str_2).numpy())
print(tf.math.equal(str_1, str_3).numpy())

输出：

False
True

从here中了解有关tf.math.equal()的更多信息。