えらく難しいオブジェクトが返ってきまくる・・・
# dataset: tensorflow.python.data.ops.dataset_ops.RangeDataset dataset = tf.data.Dataset.range(10) # dataset: tensorflow.python.data.ops.dataset_ops.WindowDataset dataset = dataset.window(5, shift=1, drop_remainder=True) # window: tensorflow.python.data.ops.dataset_ops.BatchDataset # dataset: tensorflow.python.data.ops.dataset_ops.FlatMapDataset dataset = dataset.flat_map(lambda window: window.batch(5)) # window: tensorflow.python.framework.ops.EagerTensor for window in dataset: print(window.numpy())
dataset_ops.pyで実装は見られるが・・・。
dataset_ops.py#L105-L1599が現時点での基底クラスかな。