dataset = tf.data.Dataset.from_tensor_slices(
        np.random.rand(100, 32, 32, 3)