mxnet
Public Member Functions | List of all members
mxnet::cpp::MXDataIter Class Reference

#include <io.h>

Inheritance diagram for mxnet::cpp::MXDataIter:
Inheritance graph
Collaboration diagram for mxnet::cpp::MXDataIter:
Collaboration graph

Public Member Functions

 MXDataIter (const std::string &mxdataiter_type)
 
 MXDataIter (const MXDataIter &other)
 
void BeforeFirst ()
 
bool Next ()
 
NDArray GetData ()
 
NDArray GetLabel ()
 
int GetPadNum ()
 
std::vector< int > GetIndex ()
 
MXDataIter CreateDataIter ()
 
template<typename T >
MXDataIterSetParam (const std::string &name, const T &value)
 set config parameters More...
 
- Public Member Functions inherited from mxnet::cpp::DataIter
DataBatch GetDataBatch ()
 
void Reset ()
 
virtual ~DataIter ()=default
 

Constructor & Destructor Documentation

◆ MXDataIter() [1/2]

mxnet::cpp::MXDataIter::MXDataIter ( const std::string &  mxdataiter_type)
explicit

◆ MXDataIter() [2/2]

mxnet::cpp::MXDataIter::MXDataIter ( const MXDataIter other)
inline

Member Function Documentation

◆ BeforeFirst()

void mxnet::cpp::MXDataIter::BeforeFirst ( )
virtual

Implements mxnet::cpp::DataIter.

◆ CreateDataIter()

MXDataIter mxnet::cpp::MXDataIter::CreateDataIter ( )

◆ GetData()

NDArray mxnet::cpp::MXDataIter::GetData ( )
virtual

Implements mxnet::cpp::DataIter.

◆ GetIndex()

std::vector<int> mxnet::cpp::MXDataIter::GetIndex ( )
virtual

Implements mxnet::cpp::DataIter.

◆ GetLabel()

NDArray mxnet::cpp::MXDataIter::GetLabel ( )
virtual

Implements mxnet::cpp::DataIter.

◆ GetPadNum()

int mxnet::cpp::MXDataIter::GetPadNum ( )
virtual

Implements mxnet::cpp::DataIter.

◆ Next()

bool mxnet::cpp::MXDataIter::Next ( )
virtual

Implements mxnet::cpp::DataIter.

◆ SetParam()

template<typename T >
MXDataIter& mxnet::cpp::MXDataIter::SetParam ( const std::string &  name,
const T &  value 
)
inline

set config parameters

Parameters
namename of the config parameter
valuevalue of the config parameter
Returns
reference of self

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