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