当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


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


用法

unique(
    name=None
)

参数

  • name (可选。) tf.data 操作的名称。

返回

  • 一个Dataset

丢弃 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.int32tf.int64tf.string 类型元素的数据集。

相关用法


注:本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.data.TFRecordDataset.unique。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。