如何在1D Tensor中查找重复元素

时间:2015-12-19 11:14:31

标签: python tensorflow

我想获得在1D张量中出现多次的元素。确切地说,我想创建一个与tf.unique相反的函数。例如,如果x = [1, 1, 2, 3, 4, 5, 6, 7, 4, 5, 4]我需要输出为[1,1,4,4,4,5,5],同时还要检索源张量中这些元素的索引。 我的最终目标是在批处理中获取标签出现多次的示例。

0 个答案:

没有答案