Public Member Functions | |
torch::data::Example | get (size_t i) |
Returns the example at the given index. | |
torch::optional< size_t > | size () const noexcept |
Returns the size of the dataset, or an empty optional if it is unsized. | |
Public Member Functions inherited from torch::data::datasets::Dataset< UnsizedDataset > | |
std::vector< ExampleType > | get_batch (ArrayRef< size_t > indices) override |
Returns a batch of data. More... | |
Public Member Functions inherited from torch::data::datasets::BatchDataset< UnsizedDataset, std::vector< Example<> > > | |
MapDataset< UnsizedDataset, TransformType > | map (TransformType transform)& |
Creates a MapDataset that applies the given transform to this dataset. | |
MapDataset< UnsizedDataset, TransformType > | map (TransformType transform)&& |
Creates a MapDataset that applies the given transform to this dataset. | |
Additional Inherited Members | |
Public Types inherited from torch::data::datasets::Dataset< UnsizedDataset > | |
using | ExampleType = Example<> |
Public Types inherited from torch::data::datasets::BatchDataset< UnsizedDataset, std::vector< Example<> > > | |
using | SelfType = UnsizedDataset |
using | BatchType = std::vector< Example<> > |
using | BatchRequestType = ArrayRef< size_t > |
Static Public Attributes inherited from torch::data::datasets::BatchDataset< UnsizedDataset, std::vector< Example<> > > | |
static constexpr bool | is_stateful |
Definition at line 1040 of file dataloader.cpp.