Program Listing for File data.h

↰ Return to documentation for file (torch/csrc/api/include/torch/data.h)

#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::data {
using datasets::BatchDataset; // NOLINT
using datasets::Dataset; // NOLINT
} // namespace torch::data