Public Member Functions | |
Example< torch::Tensor, std::string > | 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. | |
![]() | |
std::vector< ExampleType > | get_batch (ArrayRef< size_t > indices) override |
Returns a batch of data. More... | |
![]() | |
MapDataset< TensorStringDataset, TransformType > | map (TransformType transform)& |
Creates a MapDataset that applies the given transform to this dataset. | |
MapDataset< TensorStringDataset, TransformType > | map (TransformType transform)&& |
Creates a MapDataset that applies the given transform to this dataset. | |
Additional Inherited Members | |
![]() | |
using | ExampleType = Example< torch::Tensor, std::string > |
![]() | |
using | SelfType = TensorStringDataset |
using | BatchType = std::vector< Example< torch::Tensor, std::string > > |
using | BatchRequestType = ArrayRef< size_t > |
![]() | |
static constexpr bool | is_stateful |
Definition at line 488 of file dataloader.cpp.