Py torch 常用 的 结构

    import torch.utils.data as Data
    torch_dataset = Data.TensorDataset(x, y)	#x,y为tensor形式
    loader = Data.DataLoader(
    dataset=torch_dataset,      # torch TensorDataset format
    batch_size=BATCH_SIZE,      # mini batch size
    shuffle=True,               # 要不要打乱数据 (打乱比较好)
    num_workers=0              # 多线程来读数据，windows下要为0
    )
    for epoch in range(3):   # 训练所有!整套!数据 3 次
    for step, (batch_x, batch_y) in enumerate(loader):  # 每一步 loader 释放一小批数据用来学习