Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Types | Public Member Functions
torch::data::Iterator< Batch > Class Template Reference

Public Types

using difference_type = std::ptrdiff_t
 
using value_type = Batch
 
using pointer = Batch *
 
using reference = Batch &
 
using iterator_category = std::input_iterator_tag
 

Public Member Functions

 Iterator (std::unique_ptr< detail::IteratorImpl< Batch >> impl)
 
Iteratoroperator++ ()
 Increments the iterator. More...
 
Batch & operator* ()
 Returns the current batch. More...
 
Batch * operator-> ()
 Returns a pointer to the current batch. More...
 
bool operator== (const Iterator &other) const
 Compares two iterators for equality.
 
bool operator!= (const Iterator &other) const
 Compares two iterators for inequality.
 

Detailed Description

template<typename Batch>
class torch::data::Iterator< Batch >

Definition at line 132 of file iterator.h.

Member Function Documentation

template<typename Batch >
Batch& torch::data::Iterator< Batch >::operator* ( )
inline

Returns the current batch.

Only permitted for valid iterators (not past the end).

Definition at line 153 of file iterator.h.

template<typename Batch >
Iterator& torch::data::Iterator< Batch >::operator++ ( )
inline

Increments the iterator.

Only permitted for valid iterators (not past the end).

Definition at line 146 of file iterator.h.

template<typename Batch >
Batch* torch::data::Iterator< Batch >::operator-> ( )
inline

Returns a pointer to the current batch.

Only permitted for valid iterators (not past the end).

Definition at line 159 of file iterator.h.


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