Logo ROOT  
Reference Guide
 
Loading...
Searching...
No Matches
TMVA::DNN::TTensorBatchIterator< Data_t, Architecture_t > Class Template Reference

template<typename Data_t, typename Architecture_t>
class TMVA::DNN::TTensorBatchIterator< Data_t, Architecture_t >

TTensorBatchIterator.

Simple iterator class for the iterations over the training batches in a given data set represented by a TTensorDataLoader object.

Template Parameters
Data_tThe input data type.
Architecture_tThe underlying architecture type.

Definition at line 96 of file TensorDataLoader.h.

Public Member Functions

 TTensorBatchIterator (TTensorDataLoader< Data_t, Architecture_t > &tensorDataLoader, size_t index=0)
 
bool operator!= (const TTensorBatchIterator &other)
 
TTensorBatch< Architecture_t > operator* ()
 
TTensorBatchIterator operator++ ()
 

Private Attributes

size_t fBatchIndex
 
TTensorDataLoader< Data_t, Architecture_t > & fTensorDataLoader
 

#include <TMVA/DNN/TensorDataLoader.h>

Constructor & Destructor Documentation

◆ TTensorBatchIterator()

template<typename Data_t , typename Architecture_t >
TMVA::DNN::TTensorBatchIterator< Data_t, Architecture_t >::TTensorBatchIterator ( TTensorDataLoader< Data_t, Architecture_t > &  tensorDataLoader,
size_t  index = 0 
)
inline

Definition at line 102 of file TensorDataLoader.h.

Member Function Documentation

◆ operator!=()

template<typename Data_t , typename Architecture_t >
bool TMVA::DNN::TTensorBatchIterator< Data_t, Architecture_t >::operator!= ( const TTensorBatchIterator< Data_t, Architecture_t > &  other)
inline

Definition at line 114 of file TensorDataLoader.h.

◆ operator*()

template<typename Data_t , typename Architecture_t >
TTensorBatch< Architecture_t > TMVA::DNN::TTensorBatchIterator< Data_t, Architecture_t >::operator* ( )
inline

Definition at line 108 of file TensorDataLoader.h.

◆ operator++()

template<typename Data_t , typename Architecture_t >
TTensorBatchIterator TMVA::DNN::TTensorBatchIterator< Data_t, Architecture_t >::operator++ ( )
inline

Definition at line 109 of file TensorDataLoader.h.

Member Data Documentation

◆ fBatchIndex

template<typename Data_t , typename Architecture_t >
size_t TMVA::DNN::TTensorBatchIterator< Data_t, Architecture_t >::fBatchIndex
private

Definition at line 99 of file TensorDataLoader.h.

◆ fTensorDataLoader

template<typename Data_t , typename Architecture_t >
TTensorDataLoader<Data_t, Architecture_t>& TMVA::DNN::TTensorBatchIterator< Data_t, Architecture_t >::fTensorDataLoader
private

Definition at line 98 of file TensorDataLoader.h.

  • tmva/tmva/inc/TMVA/DNN/TensorDataLoader.h