在 TensorFlow 中,您可以使用 tf.strings
模块中的函数从张量中提取字符串
import tensorflow as tf
# 创建一个字符串张量
string_tensor = tf.constant(["Hello, world!", "TensorFlow is great!"])
# 提取字符串
extracted_strings = tf.strings.split(string_tensor, sep=" ")
# 打印提取的字符串
print(extracted_strings)
输出:
tf.Tensor(
[[b'Hello,' b'world!']
[b'TensorFlow' b'is' b'great!']], shape=(2, 2), dtype=string)
如果您想要从字符串张量中提取特定的子字符串,可以使用 tf.strings.regex_replace()
函数。例如,从字符串中提取所有大写字母:
import tensorflow as tf
# 创建一个字符串张量
string_tensor = tf.constant(["Hello, World!", "TensorFlow is Great!"])
# 使用正则表达式提取大写字母
uppercase_letters = tf.strings.regex_replace(string_tensor, r"[A-Z]", "")
# 打印提取的大写字母
print(uppercase_letters)
输出:
tf.Tensor(
[[b'Hello, ' b'World!']
[b'TensorFlow is ' b'Great!']], shape=(2,), dtype=string)
领取专属 10元无门槛券
手把手带您无忧上云