luoqiuluoqiu / WCNN

C++实现pytorch等相关功能

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

WCNN

实现pytorch主要函数

只负责实现出来,不考虑效率,错误等,只供学习所用

已经实现

  1. Conv2d
  2. MaxPool2d
  3. Relu
  4. Linear

##待实现

  • ConvTranspose2d();
  • BatchNorm2d();
  • InstanceNorm2d();
  • UpsamplingNearest2d();
  • UpsamplingBilinear2d();
  • RNN();
  • LSTM();
  • GRU();
  • Embedding();
  • Sigmoid();
ezoic increase your site revenue

About

C++实现pytorch等相关功能


Languages

Language:C++ 84.5%Language:Python 11.5%Language:QMake 4.1%