Skip to content

mbs0221-mlsys/Neural-Network

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

55 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Neural-Network

C++神经网络框架

  • 初步实现了反向传播算法,支持搭建stacked-model
  • 支持5D张量(sample, frame, width, height, channel)
  • 已实现如下基本运算类型
    • matmul/add
    • element-wise operation
    • broadcast operation
    • conv2d/conv3d operation
    • max/min/average pooling
    • max/min/average upsampling
    • permute/reshape/flatten
    • reduce_sum/reduce_mean
    • sigmoid/relu/leaky_relu/softmax
  • 初步实现了计算图框架,支持多分支网络模型
  • 预计进一步实现:
    • 基于梯度的一些优化算法

About

C++ Implementation of Neural-Network Framework

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages