gpt4 book ai didi

python - tensorflow ,删除重复项

转载 作者:太空宇宙 更新时间:2023-11-03 14:29:43 24 4
gpt4 key购买 nike

我有一个一维张量,例如:

[1,2,2,3,3,4,5]

如何从中获取[1,2,3,4,5]

我的解决方案是:

我需要一个类似于 contains 的 TensorFlow API。

然后我可以将单独的一个放入 TensorArray 中。

最佳答案

您正在寻找tf.unique。直接引用自documentation :

# tensor 'x' is [1, 1, 2, 4, 4, 4, 7, 8, 8]
y, idx = unique(x)
y ==> [1, 2, 4, 7, 8]
idx ==> [0, 0, 1, 2, 2, 2, 3, 4, 4]

关于python - tensorflow ,删除重复项,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/47385995/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com