Template Class Dataset¶
Defined in File base.h
Page Contents
Inheritance Relationships¶
Base Type¶
public torch::data::datasets::BatchDataset< Self, std::vector< Example<> > >
(Template Class BatchDataset)
Class Documentation¶
-
template<typename Self, typename SingleExample = Example<>>
class Dataset : public torch::data::datasets::BatchDataset<Self, std::vector<Example<>>>¶ A dataset that can yield data in batches, or as individual examples.
A
Dataset
is aBatchDataset
, because it supports random access and therefore batched access is implemented (by default) by calling the random access indexing function for each index in the requested batch of indices. This can be customized.Public Types
-
using ExampleType = SingleExample¶
Public Functions
-
virtual ExampleType get(size_t index) = 0¶
Returns the example at the given index.
-
inline virtual std::vector<ExampleType> get_batch(ArrayRef<size_t> indices) override¶
Returns a batch of data.
The default implementation calls
get()
for every requested index in the batch.
-
using ExampleType = SingleExample¶