我有一个4D数组形式的图像数据，其中包含841个图像，每个图像大56x56像素，并具有33个“彩色”通道。现在，我有一个用于低分辨率图像的4D阵列（从高分辨率图像中降采样）和一个用于高分辨率图像的4D阵列，它们应该用作损失函数的基础。 现在我的问题是如何使用这些数组创建数据集，以便拥有可以与以下代码一起使用的LR-HR图像对（最初在Div2K上试用时使用的代码）：
train = ... #dataset with HR-LR pairs
train_ds = train.dataset(batch_size=16, random_transform=True)
model.compile(optimizer='adam', loss='mean_absolute_error')
model.fit(train_ds, epochs=300, steps_per_epoch=1000)