无法在tensorflow中获取数据集的形状属性值

Question

您可以检查以下代码，

import tensorflow as tf
data = tf.data.Dataset.range(10)
tf.print(data)

输出为

<RangeDataset shapes: (), types: tf.int64>

该形状为空。

Answer 1

就像python的range一样，Dataset.range也不返回实际值。相反，它返回一个类似生成器的对象RangeDataset。要获得numpy迭代器，您需要RangeDataset.as_numpy_iterator。然后，您可以将其转换为列表，就像使用list(range(5)):一样

>>> tf.data.Dataset.range(5)
<RangeDataset shapes: (), types: tf.int64>
>>> list(tf.data.Dataset.range(5).as_numpy_iterator())
[0, 1, 2, 3, 4]
>>> range(5)
range(0, 5)
>>> list(range(5))
[0, 1, 2, 3, 4]

有关其用法的更多示例，您可以查看文档