如何为不同通道构建具有多个数组的PyTorch数据集

我希望知道如何使用多个NumPy数组作为数据通道来构建数据集。 我有多个阵列，4通道数据作为输入，1通道数据作为输出。 例如，

我认为这种形式的数据可以通过以下功能构建

from torch.utils.data.dataset import Dataset

但是我仍然有一些问题需要正确分配张量，指定输入和输出。 我希望能得到一些关于这个问题的想法或指导

非常感谢你