train_iter = torch.utils.data.DataLoader(mnist_train, batch_size=batch_size, shuffle=True, num_workers=num_workers)

这是用来做什么的？

这行代码是用来创建一个 PyTorch 的 DataLoader 对象，用于将数据集 mnist_train 分批次地读取到模型中进行训练。其中，batch_size 参数设置每个批次的数据量，shuffle 参数表示是否打乱数据集顺序，num_workers 参数表示使用多少个进程来并行载入数据。