data_loader = torch.utils.data.DataLoader(dataset=mnist_train,
                                          batch_size=batch_size,
                                          shuffle=True,
                                          drop_last=True)

dataset : 어떤 데이터를 사용하는가

batch_size : batch_size크기

shuffle : 데이터를 섞을 것인가?

drop_last : batch_size씩하고 남은 데이터 자를래?

사용법

for epoch in range(training_epochs):
    '''
    for X, Y in data_loader:

X: MNIST이미지
Y: Label