InfiniteStreamDataset Struct Reference

Inheritance diagram for InfiniteStreamDataset:

Public Member Functions
std::vector< int >	get_batch (size_t batch_size) override
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::BatchDataset< Self, Batch, BatchRequest >
virtual Batch	get_batch (BatchRequest request)=0
	Returns a batch of data given an index.
template<typename TransformType >
MapDataset< Self, TransformType >	map (TransformType transform)&
	Creates a MapDataset that applies the given transform to this dataset.
template<typename TransformType >
MapDataset< Self, TransformType >	map (TransformType transform)&&
	Creates a MapDataset that applies the given transform to this dataset.

Data Fields
size_t	counter = 0

Additional Inherited Members
Public Types inherited from torch::data::datasets::BatchDataset< Self, Batch, BatchRequest >
using	SelfType = Self
using	BatchType = Batch
using	BatchRequestType = BatchRequest
Static Public Attributes inherited from torch::data::datasets::BatchDataset< Self, Batch, BatchRequest >
static constexpr bool	is_stateful = detail::is_optional<BatchType>::value

Detailed Description

Definition at line 133 of file dataloader.cpp.

---

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

• test/cpp/api/dataloader.cpp