【python】pytorch中如何使用DataLoader对数据集进行批处理

第一步: 我们要创建torch能够识别的数据集类型(pytorch中也有很多现成的数据集类型,以后再说)。 首先我们建立两个向量X和Y,一个作为输入的数据,一个作为正确的结果:      随后我们需要把X和Y组成一个完整的数据集,并转化为pytorch能识别的数据集类型:      我们来看一下这些数据的数据类型:       可以看出我们把X和Y通过Data.TensorDataset() 这个
相关文章
相关标签/搜索