一个基于C++和python的张量计算库,目标是用C++实现常见的张量操作,便于将python的原型代码改写成C++,以及用C++实现卷积操作。目前已经实现的功能包括 多维张量加法减法乘法,relu,flatten,张量变形(NHWC => NCHW),张量线性计算(xw+b), 最大池化,后续还将实现二维卷积,并添加CNN训练部分的代码。...

新闻来源:Gitee 推荐 | 基于 C++ 和 python 的张量计算库 tensor-calcu-lib