import os

os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"

import tensorflow as tf

layers = tf.keras.layers

tf.get_logger().setLevel("ERROR")
tf.keras.utils.set_random_seed(7)

IMAGE_SIZE = 64
BATCH_SIZE = 4

images = tf.reshape(
    tf.range(8 * 48 * 48 * 3, dtype=tf.float32),
    (8, 48, 48, 3),
)
images = tf.math.floormod(images, 256.0)
labels = tf.constant([0, 1, 0, 1, 0, 1, 0, 1], dtype=tf.int32)