Pytorch Dataloader Example

gistfile1.txt

import torch
from torch.utils.data import DataLoader, TensorDataset

# Define the dataset
X = torch.Tensor([[1, 2], [3, 4], [5, 6], [7, 8]])
y = torch.Tensor([0, 1, 0, 1])
dataset = TensorDataset(X, y)

# Create the dataloader
dataloader = DataLoader(dataset, batch_size=2, shuffle=True)

# Iterate through the dataloader and process the data
for i, (inputs, labels) in enumerate(dataloader):
  print(f'Batch {i}:')
  print(f'  Inputs: {inputs}')
  print(f'  Labels: {labels}')