import torch
from torch.utils import data
data_arrays=(features, labels)
dataset = data.TensorDataset(*data_arrays)
data_iter = data.DataLoader(dataset, batch_size, shuffle=True)
TensorDataset
最新推荐文章于 2024-01-23 13:36:05 发布
import torch
from torch.utils import data
data_arrays=(features, labels)
dataset = data.TensorDataset(*data_arrays)
data_iter = data.DataLoader(dataset, batch_size, shuffle=True)