當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文

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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。