TensorFlow Datasets 数据集载入

我想请教下,map 这里如果不使用 lambda 表达式的话,直接使用 函数 ,需要怎么写函数?
def preprocess (……):
……
return ……

原文代码如下:

使用 TessorFlow Datasets 载入 “tf_flowers” 数据集

dataset = tfds.load (“tf_flowers”, split=tfds.Split.TRAIN, as_supervised=True)

对 dataset 进行大小调整、打散和分批次操作

dataset = dataset.map (lambda img, label: (tf.image.resize (img, [224, 224]) / 255.0, label))
.shuffle (1024)
.batch (32)

迭代数据

for images, labels in dataset:
# 对 images 和 labels 进行操作