我正在尝试在tensorflow中完成所有输入管道的工作。这包括将示例转换为分类器所需的类型。
我刚刚了解到我无法像使用标准python列表那样遍历字符串张量。我的具体问题是“是否有一个tf函数来测试张量中是否存在恒定值?”当然,可能会有更好的方法(我是tf和python的新手)。
# creating a unique list of tokens (python)
a_global = []
a = [...]
for token in a:
if a_global.count(token) == 0:
a_global.append(token)
我正在为字符串标记编制索引,因此我基本上可以使用标记在列表中的位置作为新值将它们转换为整数。当“ a”是张量时,该代码段将不起作用,因此我正在尝试使用tf.map_fn(),但我不知道如何复制IF语句谓词。有人可以指出我正确的方向吗?
tf版本1.8