TensorFlowのDatasetを知る - Qiita

ds = ds.shuffle(count) ds = ds.batch(batch_size, drop_remainder=True) ds = ds.prefetch(buffer_size=AUTOTUNE) テスト用はshuffleをぬいて、drop_remainderをFalseにする。 (drop_remainderはpytorchで言えばdrop_lastと同じものです) やりたいこと 大抵のニューラルネットワークの学習時には1エポックで、下記のようにバッチを取得して学習したいのではないでしょうか。下記は全20データを3バッチずつとってくるイメージ。これをtensorflowのdsではどうやったらできるか？というのが今回の目的。2021年に書くことじゃないだろ、と思うんですが、最近知ったこともあるので備忘録として・・ この記事が機能の詳細は、かなり細かく載せていますので、参