1#pragma once
2
3#include <torch/data/dataloader.h>
4#include <torch/data/datasets.h>
5#include <torch/data/samplers.h>
6#include <torch/data/transforms.h>
7
8// Some "exports".
9namespace torch {
10namespace data {
11using datasets::BatchDataset;
12using datasets::Dataset;
13} // namespace data
14} // namespace torch
15