Okay can you try this in the for loop-
ctr = 0
for idx, (batch, target) in enumerate(train_loader):
    if ctr ==0:
        print(batch)
    batch += 1

Now check if it shuffles it?
I did this and it didn’t shuffle along feature dimension for me