The DataLoader uses the provided Dataset to create the batches. Based on your description it seems that you would like to add the new tensor to the Dataset. Outside of the DataLoader loop, you could access it via dataloader.dataset and could also try to reassign a new Dataset. However, the TensorDataset doesn’t provide a method to add new tensors, so you would have to recreate it.