pytorch生成带滑动的训练集测试集(batch)
话不多说,先上代码
Example1
from torch.utils import data
import torch
batch_size = 3
t1 = np.arange(0,20)
t2 = np.arange(0,20)+100
生成序列0~19和100~190,分别模拟x_train和y_train
tensor1 = torch.from_numpy(t1)
tensor2 = torch.from_numpy(t2)
上一步很关键,否则报错
data_set = da
原创
2022-04-14 17:16:46 ·
1547 阅读 ·
0 评论