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