C++学习笔记(二)

建立NDArray,访问元素

std::vector<mx_float> v {1.23, 4.56, 7.89, 5.71};
mxnet::cpp::NDArray nda_array {v, Shape{2,2}, Context::cpu()};

assert(nda_array.Size() == 4);
assert(nda_array.At(0, 0) == v[0]);
assert(nda_array.At(0, 1) == v[1]);
assert(nda_array.At(1, 0) == v[2]);
assert(nda_array.At(1, 1) == v[3]);

---

批量抓取文件

#include <corecrt_io.h>

std::string path = "D:/Workspace/Image";
path = path + "/*.png";

_finddata_t file;//_finddata_t为文件结构体
intptr_t handle;
std::string name;

handle = _findfirst(path.c_str(), &file);
if (handle == -1L) //若目录下目标文件为空，_findfirst()函数返回-1L
{
	cout << "no files!!! please check the dir" << endl;
	return 0;
}
else
{
	do
	{
		name = file.name;
		cout << name << endl;
	} while (_findnext(handle, &file) == 0);//_findnext()函数若查找成功，返回0，失败返回-1

}//else
_findclose(handle);

参考链接

---