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

Python tf.data.experimental.cardinality用法及代码示例


返回 dataset 的基数(如果已知)。

用法

tf.data.experimental.cardinality(
    dataset
)

参数

返回

  • 一个标量 tf.int64 Tensor 表示 dataset 的基数。如果基数是无限的或未知的，则操作分别返回命名常量INFINITE_CARDINALITY 和UNKNOWN_CARDINALITY。

该操作返回 dataset 的基数。如果 dataset 包含无限数量的元素，则该操作可能会返回 tf.data.experimental.INFINITE_CARDINALITY，如果分析无法确定 dataset 中的元素数量(例如，当数据集源是文件时)，则该操作可能会返回 tf.data.experimental.UNKNOWN_CARDINALITY。

dataset = tf.data.Dataset.range(42)
print(tf.data.experimental.cardinality(dataset).numpy())
42
dataset = dataset.repeat()
cardinality = tf.data.experimental.cardinality(dataset)
print((cardinality == tf.data.experimental.INFINITE_CARDINALITY).numpy())
True
dataset = dataset.filter(lambda x:True)
cardinality = tf.data.experimental.cardinality(dataset)
print((cardinality == tf.data.experimental.UNKNOWN_CARDINALITY).numpy())
True

相关用法


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