Hi Prerna,
I am loading the data from a 42000x2048 array, and then creating a 16x1x2048 tensor called data, when i view this it has lost the order from the 42000x2048 array
train_dataset = TensorDataset(X_train, y_train)
train_loader = DataLoader(train_dataset, batch_size=BATCHSIZE, shuffle=False)

for epoch in range(num_epochs):
    for data, target in train_loader:
#        print("data=",data.shape)
        np.savetxt(f,data.numpy())
        np.savetxt(f2,target.numpy())
        data=data.unsqueeze(1)
        print("data_sq=",data.shape)
        data = data.cuda()

I think its shuffling along the feature, though looking at the data i would guess that its generating a 16x1x2048 tensor of numbers randomly taken from x_train array