no_data_loss.py

import tensorflow as tf
from functools import partial

ds_raw = tf.constant([i for i in range(0, 100)])
ds = tf.data.Dataset.from_tensor_slices(ds_raw)

ds_a = ds.window(2).flat_map(lambda x: x.batch(2))
# for b, skip 1 and keep only last element
ds_b = ds.skip(1).window(2).flat_map(lambda x: x.batch(2)).map(lambda x: x[-1])

for a, b in tf.data.Dataset.zip((ds_a, ds_b)):
    print(f'a: {a} -> b: {b}')
    
# a: [0 1] -> b: 2
# a: [2 3] -> b: 4