Public Member Functions | |
Example< torch::Tensor, int > | get (size_t index) override |
Returns the example at the given index. | |
torch::optional< size_t > | size () const override |
Returns the size of the dataset, or an empty optional if it is unsized. | |
Public Member Functions inherited from torch::data::datasets::Dataset< DummyTensorDataset, Example< torch::Tensor, int > > | |
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< DummyTensorDataset, std::vector< Example< torch::Tensor, int > > > | |
MapDataset< DummyTensorDataset, TransformType > | map (TransformType transform)& |
Creates a MapDataset that applies the given transform to this dataset. | |
MapDataset< DummyTensorDataset, 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< DummyTensorDataset, Example< torch::Tensor, int > > | |
using | ExampleType = Example< torch::Tensor, int > |
Public Types inherited from torch::data::datasets::BatchDataset< DummyTensorDataset, std::vector< Example< torch::Tensor, int > > > | |
using | SelfType = DummyTensorDataset |
using | BatchType = std::vector< Example< torch::Tensor, int > > |
using | BatchRequestType = ArrayRef< size_t > |
Static Public Attributes inherited from torch::data::datasets::BatchDataset< DummyTensorDataset, std::vector< Example< torch::Tensor, int > > > | |
static constexpr bool | is_stateful |
Definition at line 525 of file dataloader.cpp.