torch_frame.data.DataLoader

class DataLoader(dataset: Dataset | TensorFrame, *args, **kwargs)

基础类：DataLoader

一个数据加载器，它从torch_frame.Dataset或torch_frame.TensorFrame对象创建小批量数据。

import torch_frame

dataset = ...

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

Parameters:

• dataset (Dataset 或 TensorFrame) – 从中加载数据的数据集或张量框架。

• *args (可选) – torch.utils.data.DataLoader 的额外参数。

• **kwargs (可选) – torch.utils.data.DataLoader 的额外关键字参数。