python - Tensorflow有类似python中set()的数据类型吗？

Question

我想实现一个提取 Tensorflow 张量中所有不同元素的操作。在python中，我通常使用set(array)，但我不知道Tensorflow中是否有像set这样的数据类型或任何功能相同，效果相同。

Answer 1

要从张量中获取所有唯一元素，请使用tf.unique()。首先，将张量 reshape 为 rank=1，然后应用 tf.unique():

import tensorflow as tf

tensor = tf.constant([[1., 2., 3.], [3., 4., 5.]], dtype=tf.float32)
flatten = tf.reshape(tensor, [-1]) 
unique = tf.unique(flatten)[0]

with tf.Session() as sess:
    print(unique.eval()) # [1. 2. 3. 4. 5.]