I’m not sure what make_one_shot_iterator does, as the docs only mention:

Creates an iterator for elements of dataset .

If you want to iterate the PyTorch Dataset, you could just use it in a loop:
dataset = datasets.MNIST(root='PATH', transform=...)

for data, target in dataset:
    # ...