torch_frame.data.DataLoader

class DataLoader(dataset: torch_frame.data.dataset.Dataset | torch_frame.data.tensor_frame.TensorFrame, *args, **kwargs)

Bases: DataLoader

A data loader which creates mini-batches from a torch_frame.Dataset or torch_frame.TensorFrame object.

import torch_frame

dataset = ...

loader = torch_frame.data.DataLoader(
    dataset,
    batch_size=512,
    shuffle=True,
)

Parameters:

• dataset (Dataset or TensorFrame) – The dataset or tensor frame from which to load the data.

• *args (optional) – Additional arguments of torch.utils.data.DataLoader.

• **kwargs (optional) – Additional keyword arguments of torch.utils.data.DataLoader.