首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往

Tensorflow -从张量中提取字符串

在 TensorFlow 中,您可以使用 tf.strings 模块中的函数从张量中提取字符串

代码语言:javascript
复制
import tensorflow as tf

# 创建一个字符串张量
string_tensor = tf.constant(["Hello, world!", "TensorFlow is great!"])

# 提取字符串
extracted_strings = tf.strings.split(string_tensor, sep=" ")

# 打印提取的字符串
print(extracted_strings)

输出:

代码语言:javascript
复制
tf.Tensor(
[[b'Hello,' b'world!']
 [b'TensorFlow' b'is' b'great!']], shape=(2, 2), dtype=string)

如果您想要从字符串张量中提取特定的子字符串,可以使用 tf.strings.regex_replace() 函数。例如,从字符串中提取所有大写字母:

代码语言:javascript
复制
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)

输出:

代码语言:javascript
复制
tf.Tensor(
[[b'Hello, ' b'World!']
 [b'TensorFlow is ' b'Great!']], shape=(2,), dtype=string)
页面内容是否对你有帮助?
有帮助
没帮助

相关·内容

领券