Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
TensorStringDataset Struct Reference
Inheritance diagram for TensorStringDataset:
torch::data::datasets::Dataset< TensorStringDataset, Example< torch::Tensor, std::string > > torch::data::datasets::BatchDataset< TensorStringDataset, std::vector< Example< torch::Tensor, std::string > > >

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.
 
- Public Member Functions inherited from torch::data::datasets::Dataset< TensorStringDataset, Example< torch::Tensor, std::string > >
std::vector< ExampleTypeget_batch (ArrayRef< size_t > indices) override
 Returns a batch of data. More...
 
- Public Member Functions inherited from torch::data::datasets::BatchDataset< TensorStringDataset, std::vector< Example< torch::Tensor, std::string > > >
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

- Public Types inherited from torch::data::datasets::Dataset< TensorStringDataset, Example< torch::Tensor, std::string > >
using ExampleType = Example< torch::Tensor, std::string >
 
- Public Types inherited from torch::data::datasets::BatchDataset< TensorStringDataset, std::vector< Example< torch::Tensor, std::string > > >
using SelfType = TensorStringDataset
 
using BatchType = std::vector< Example< torch::Tensor, std::string > >
 
using BatchRequestType = ArrayRef< size_t >
 
- Static Public Attributes inherited from torch::data::datasets::BatchDataset< TensorStringDataset, std::vector< Example< torch::Tensor, std::string > > >
static constexpr bool is_stateful
 

Detailed Description

Definition at line 488 of file dataloader.cpp.


The documentation for this struct was generated from the following file: