Go to the documentation of this file.
25 #ifndef MXNET_CPP_IO_H_
26 #define MXNET_CPP_IO_H_
34 #include "dmlc/logging.h"
52 virtual bool Next(
void) = 0;
56 virtual std::vector<int>
GetIndex(
void) = 0;
71 mx_uint num_data_iter_creators = 0;
75 for (
mx_uint i = 0; i < num_data_iter_creators; i++) {
77 const char* description;
79 const char** arg_names;
80 const char** arg_type_infos;
81 const char** arg_descriptions;
90 mxdataiter_creators_[name] = data_iter_creators[i];
94 return mxdataiter_creators_[name];
98 std::map<std::string, DataIterCreator> mxdataiter_creators_;
116 explicit MXDataIter(
const std::string& mxdataiter_type);
118 creator_ = other.creator_;
119 params_ = other.params_;
120 blob_ptr_ = other.blob_ptr_;
135 template <
typename T>
137 std::string value_str;
138 std::stringstream ss;
142 params_[name] = value_str;
148 std::map<std::string, std::string> params_;
149 std::shared_ptr<MXDataIterBlob> blob_ptr_;
155 #endif // MXNET_CPP_IO_H_
namespace of mxnet
Definition: api_registry.h:33
MXDataIter(const std::string &mxdataiter_type)
MXDataIter(const MXDataIter &other)
Definition: io.h:117
Default object for holding a mini-batch of data and related information.
Definition: io.h:42
DataBatch GetDataBatch()
Definition: io.h:58
virtual std::vector< int > GetIndex(void)=0
~MXDataIterBlob()
Definition: io.h:105
MXNET_DLL int MXDataIterGetIterInfo(DataIterCreator creator, const char **name, const char **description, uint32_t *num_args, const char ***arg_names, const char ***arg_type_infos, const char ***arg_descriptions)
Get the detailed information about data iterator.
MXDataIterMap()
Definition: io.h:70
MXDataIter & SetParam(const std::string &name, const T &value)
set config parameters
Definition: io.h:136
MXDataIterBlob()
Definition: io.h:103
NDArray interface.
Definition: ndarray.h:122
virtual NDArray GetData(void)=0
DataIterHandle handle_
Definition: io.h:108
virtual bool Next(void)=0
void * DataIterHandle
handle to a DataIterator
Definition: c_api.h:90
void Reset()
Definition: io.h:61
MXDataIter CreateDataIter()
void * DataIterCreator
handle a dataiter creator
Definition: c_api.h:88
MXNET_DLL int MXListDataIters(uint32_t *out_size, DataIterCreator **out_array)
List all the available iterator entries.
NDArray data
Definition: io.h:44
int pad_num
Definition: io.h:46
MXDataIterBlob(DataIterHandle handle)
Definition: io.h:104
std::vector< int > index
Definition: io.h:47
virtual NDArray GetLabel(void)=0
virtual void BeforeFirst(void)=0
base definitions for mxnetcpp
MXNET_DLL int MXDataIterFree(DataIterHandle handle)
Free the handle to the IO module.
DataIterCreator GetMXDataIterCreator(const std::string &name)
Definition: io.h:93
uint32_t mx_uint
manually define unsigned int
Definition: c_api.h:65
NDArray label
Definition: io.h:45
std::vector< int > GetIndex()
virtual int GetPadNum(void)=0
virtual ~DataIter()=default