Python tf.data.TFRecordDataset.unique用法及代碼示例

用法

unique(
    name=None
)

丟棄 Dataset 的重複元素的轉換。

使用此轉換生成一個數據集，該數據集包含輸入中每個唯一元素的一個實例。例如：

dataset = tf.data.Dataset.from_tensor_slices([1, 37, 2, 37, 2, 1])
dataset = dataset.unique()
sorted(list(dataset.as_numpy_iterator()))
[1, 2, 37]

注意：此轉換僅支持適合內存且具有 tf.int32 、 tf.int64 或 tf.string 類型元素的數據集。