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

Question

我希望知道如何使用多个 NumPy 数组作为数据通道来构建数据集。 我有多个数组用于 4 通道数据作为输入和 1 通道数据作为输出。 例如， Example of Data Structure - Tensor Form

我认为可以使用以下函数构建这种形式的数据，

from torch.utils.data.dataset import Dataset

但是我在正确分配张量、指定输入和输出方面仍然存在一些问题。 我希望得到一些解决这个问题的想法或指导。

非常感谢！